| H A D | core_named_ops.py | 6 Batch = S.Batch variable 120 def batch_matmul(A=TensorDef(T1, Batch, S.M, S.K), argument 121 B=TensorDef(T2, Batch, S.K, S.N), 122 C=TensorDef(U, Batch, S.M, S.N, output=True)): 136 def quantized_batch_matmul(A=TensorDef(T1, Batch, S.M, S.K), argument 137 B=TensorDef(T2, Batch, S.K, S.N), 140 C=TensorDef(U, Batch, S.M, S.N, output=True)): 183 def batch_matvec(A=TensorDef(T1, Batch, S.M, S.K), argument 184 B=TensorDef(T2, Batch, S.K), 185 C=TensorDef(U, Batch, S.M, output=True)):
|