Searched refs:transpose_b_ (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_matmul_op.cc | 72 OP_REQUIRES_OK(context, context->GetAttr("transpose_b", &transpose_b_)); in QuantizedMatMulOp()
103 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
141 meta::QuantizedGemm(context, transpose_a_, transpose_b_, a_data, b_data, in Compute()
150 if (transpose_b_) { in Compute()
160 if (transpose_b_) { in Compute()
172 transpose_a_, transpose_b_, transpose_c, m, n, k, a_data, offset_a, in Compute()
191 bool transpose_b_; member in tensorflow::QuantizedMatMulOp
|
| D | mkl_matmul_op.cc | 49 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_b", &transpose_b_)); in MklMatMulOp()
63 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
108 bool transpose_b_; member in tensorflow::MklMatMulOp
|
| D | matmul_op.cc | 444 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_b", &transpose_b_)); in MatMulOp()
466 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
527 bool transpose_b_; member in tensorflow::MatMulOp
|
| D | sparse_matmul_op.cc | 962 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_b", &transpose_b_)); in SparseMatMulOp()
977 const int n = transpose_b_ ? b.dim_size(0) : b.dim_size(1); in Compute()
978 const int k2 = transpose_b_ ? b.dim_size(1) : b.dim_size(0); in Compute()
1017 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
1028 bool transpose_b = transpose_b_; in Compute()
1073 bool transpose_b_; member in tensorflow::SparseMatMulOp
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matmul_op.cc | 35 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_b", &transpose_b_)); in MatMulOp()
66 int second_index = transpose_b_ ? 1 : 0; in Compile()
85 auto rhs = (transpose_b_) ? xla::Transpose(b, {1, 0}) : b; in Compile()
92 bool transpose_b_; member in tensorflow::__anondc94a7a40111::MatMulOp
|