jax.experimental.pallas.dot

jax.experimental.pallas.dot(a, b, trans_a=False, trans_b=False, allow_tf32=None, precision=None)

Parameters:

• trans_a (bool)

• trans_b (bool)

• allow_tf32 (bool | None)