H A D | core_named_ops.py | 6 Batch = S.Batch variable 398 includes zero-point adjustments for the left and right operands of the 402 C[D.m, D.n] += (TypeFn.cast_signed(U, A[D.m, D.k]) - TypeFn.cast_signed(U, AZp)) * ( 403 TypeFn.cast_signed(U, B[D.k, D.n]) - TypeFn.cast_signed(U, BZp) 449 """Performs a matrix-matrix-transpose multiplication of two 4D inputs. 468 lhs=TensorDef(TV.LhsType, Batch, S.M, S.K, S.M0, S.K0), 469 rhs=TensorDef(TV.RhsType, Batch, S.N, S.K, S.N0, S.K0), 470 accum=TensorDef(TV.AccumType, Batch, 491 batch_mmt4d( lhs=TensorDef(TV.LhsType, Batch, S.M, S.K, S.M0, S.K0), rhs=TensorDef(TV.RhsType, Batch, S.N, S.K, S.N0, S.K0), accum=TensorDef(TV.AccumType, Batch, S.M, S.N, S.M0, S.N0, output=True), ) global() argument 512 batch_matmul( A=TensorDef(T1, Batch, S.M, S.K), B=TensorDef(T2, Batch, S.K, S.N), C=TensorDef(U, Batch, S.M, S.N, output=True), ) global() argument 530 batch_matmul_transpose_a( A=TensorDef(T1, Batch, S.K, S.M), B=TensorDef(T2, Batch, S.K, S.N), C=TensorDef(U, Batch, S.M, S.N, output=True), ) global() argument 549 batch_matmul_transpose_b( A=TensorDef(T1, Batch, S.M, S.K), B=TensorDef(T2, Batch, S.N, S.K), C=TensorDef(U, Batch, S.M, S.N, output=True), ) global() argument 568 quantized_batch_matmul( A=TensorDef(T1, Batch, S.M, S.K), B=TensorDef(T2, Batch, S.K, S.N), AZp=ScalarDef(I32), BZp=ScalarDef(I32), C=TensorDef(U, Batch, S.M, S.N, output=True), ) global() argument 589 batch_reduce_matmul( A=TensorDef(T1, Batch, S.M, S.K), B=TensorDef(T2, Batch, S.K, S.N), C=TensorDef(U, S.M, S.N, output=True), ) global() argument 636 batch_matvec( A=TensorDef(T1, Batch, S.M, S.K), B=TensorDef(T2, Batch, S.K), C=TensorDef(U, Batch, S.M, output=True), ) global() argument 654 batch_vecmat( A=TensorDef(T1, Batch, S.K), B=TensorDef(T2, Batch, S.K, S.N), C=TensorDef(U, Batch, S.N, output=True), ) global() argument [all...] |