| /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()
636 if (!this->transpose_b_) { in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matmul_op.cc | 38 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_b", &transpose_b_)); in MatMulOp()
69 int second_index = transpose_b_ ? 1 : 0; in Compile()
87 ctx->SetOutput(0, xla::BatchDot(a, transpose_a_, b, transpose_b_)); in Compile()
93 bool transpose_b_; member in tensorflow::__anondc94a7a40111::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 | matmul_op_fused.cc | 153 OP_REQUIRES_OK(context, context->GetAttr("transpose_b", &transpose_b_)); in FusedMatMulOp()
188 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
224 bool transpose_b_; member in tensorflow::FusedMatMulOp
|
| 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
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op.cc | 45 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_b", &transpose_b_)); in MklMatMulOp()
59 dim_pair[0].second = transpose_b_ ? 1 : 0; in Compute()
105 bool transpose_b_; member in tensorflow::MklMatMulOp
|
| D | mkl_matmul_op_fused.cc | 37 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_b", &transpose_b_)); in MklFusedMatMulOp()
88 const int dim_pair[] = {1, transpose_b_ ? 1 : 0}; in Compute()
114 transpose_b_ ? memory::format_tag::oi : memory::format_tag::io; in Compute()
294 bool transpose_b_; member in tensorflow::MklFusedMatMulOp
|