| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_mat_mul_op.cc | 92 OP_REQUIRES_OK(c, c->GetAttr("transpose_b", &transpose_b_)); in CSRSparseMatMulCPUOp()
98 OP_REQUIRES(c, !(adjoint_b_ && transpose_b_), in CSRSparseMatMulCPUOp()
141 if (transpose_b_ || adjoint_b_) in Compute()
194 transpose_b_, adjoint_b_); in Compute()
287 bool transpose_b_; member in tensorflow::CSRSparseMatMulCPUOp
297 OP_REQUIRES_OK(c, c->GetAttr("transpose_b", &transpose_b_)); in CSRSparseMatMulGPUOp()
305 OP_REQUIRES(c, !(adjoint_b && transpose_b_), in CSRSparseMatMulGPUOp()
311 transpose_b_ = transpose_b_ || adjoint_b; in CSRSparseMatMulGPUOp()
348 b_tensor_shape.dim_size(transpose_b_ ? row_dim + 1 : row_dim); in Compute()
366 b_tensor_shape.dim_size(transpose_b_ ? row_dim : row_dim + 1); in Compute()
[all …]
|
| D | mat_mul_op.cc | 83 OP_REQUIRES_OK(c, c->GetAttr("transpose_b", &transpose_b_)); in CSRMatMulOp()
91 OP_REQUIRES(c, !(adjoint_b && transpose_b_), in CSRMatMulOp()
99 transpose_b_ |= adjoint_b; in CSRMatMulOp()
130 dense_tensor_b.dim_size(this->transpose_b_ ? *rank - 1 : *rank - 2); in ValidateInputs()
142 bool transpose_b_; member in tensorflow::CSRMatMulOp
192 if (this->transpose_b_) { in Compute()
519 b_t.shape().dim_size(this->transpose_b_ ? row_dim + 1 : row_dim); in Compute()
521 b_t.dim_size(this->transpose_b_ ? row_dim : row_dim + 1); in Compute()
631 if (!this->transpose_b_) { in Compute()
|
| /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::__anon7b49604f0111::MatMulOp
|
| /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_fused.cc | 37 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_b", &transpose_b_)); in MklFusedMatMulOp()
82 const int dim_pair[] = {1, transpose_b_ ? 1 : 0}; in Compute()
107 transpose_b_ ? memory::format::oi : memory::format::io; in Compute()
202 bool transpose_b_; member in tensorflow::MklFusedMatMulOp
|
| D | matmul_op_fused.cc | 110 OP_REQUIRES_OK(context, context->GetAttr("transpose_b", &transpose_b_)); in FusedMatMulOp()
142 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
178 bool transpose_b_; member in tensorflow::FusedMatMulOp
|
| D | mkl_matmul_op.cc | 50 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_b", &transpose_b_)); in MklMatMulOp()
64 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
109 bool transpose_b_; member in tensorflow::MklMatMulOp
|
| D | matmul_op.cc | 447 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_b", &transpose_b_)); in MatMulOp()
469 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
530 bool transpose_b_; member in tensorflow::MatMulOp
|
| D | sparse_matmul_op.cc | 961 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_b", &transpose_b_)); in SparseMatMulOp()
976 const int n = transpose_b_ ? b.dim_size(0) : b.dim_size(1); in Compute()
977 const int k2 = transpose_b_ ? b.dim_size(1) : b.dim_size(0); in Compute()
1016 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
1027 bool transpose_b = transpose_b_; in Compute()
1072 bool transpose_b_; member in tensorflow::SparseMatMulOp
|