| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_mat_mul_op.cc | 91 OP_REQUIRES_OK(c, c->GetAttr("transpose_a", &transpose_a_)); in CSRSparseMatMulCPUOp()
94 OP_REQUIRES(c, !(adjoint_a_ && transpose_a_), in CSRSparseMatMulCPUOp()
139 if (transpose_a_ || adjoint_a_) in Compute()
192 transpose_a_, adjoint_a_); in Compute()
286 bool transpose_a_; member in tensorflow::CSRSparseMatMulCPUOp
296 OP_REQUIRES_OK(c, c->GetAttr("transpose_a", &transpose_a_)); in CSRSparseMatMulGPUOp()
300 OP_REQUIRES(c, !(adjoint_a && transpose_a_), in CSRSparseMatMulGPUOp()
310 transpose_a_ = transpose_a_ || adjoint_a; in CSRSparseMatMulGPUOp()
346 a_tensor_shape.dim_size(transpose_a_ ? row_dim : row_dim + 1); in Compute()
364 a_tensor_shape.dim_size(transpose_a_ ? row_dim + 1 : row_dim); in Compute()
[all …]
|
| D | mat_mul_op.cc | 82 OP_REQUIRES_OK(c, c->GetAttr("transpose_a", &transpose_a_)); in CSRMatMulOp()
86 OP_REQUIRES(c, !(adjoint_a && transpose_a_), in CSRMatMulOp()
98 transpose_a_ |= adjoint_a; in CSRMatMulOp()
128 a_dense_shape(this->transpose_a_ ? *rank - 2 : *rank - 1); in ValidateInputs()
141 bool transpose_a_; member in tensorflow::CSRMatMulOp
185 if (this->transpose_a_) { in Compute()
210 if (!this->transpose_a_) { in Compute()
517 this->transpose_a_ ? row_dim + 1 : row_dim); in Compute()
545 use_matrix_vector_multiply && !this->transpose_a_; in Compute()
571 functor::CSRSparseMatrixMatVec<GPUDevice, T> csr_spmv(this->transpose_a_, in Compute()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matmul_op.cc | 37 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_a", &transpose_a_)); in MatMulOp()
68 int first_index = transpose_a_ ? 0 : 1; in Compile()
87 ctx->SetOutput(0, xla::BatchDot(a, transpose_a_, b, transpose_b_)); in Compile()
92 bool transpose_a_; member in tensorflow::__anondc94a7a40111::MatMulOp
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | triangular_solve_thunk.cc | 55 transpose_a_ = [&] { in TriangularSolveThunk()
99 .ThenBlasTrsm(side_, uplo_, transpose_a_, in ExecuteOnStream()
109 .ThenBlasTrsm(side_, uplo_, transpose_a_, in ExecuteOnStream()
120 .ThenBlasTrsm(side_, uplo_, transpose_a_, unit_diagonal_, m_, in ExecuteOnStream()
131 .ThenBlasTrsm(side_, uplo_, transpose_a_, unit_diagonal_, m_, in ExecuteOnStream()
|
| D | triangular_solve_thunk.h | 57 se::blas::Transpose transpose_a_; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_matmul_op.cc | 71 OP_REQUIRES_OK(context, context->GetAttr("transpose_a", &transpose_a_)); in QuantizedMatMulOp()
102 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
141 meta::QuantizedGemm(context, transpose_a_, transpose_b_, a_data, b_data, in Compute()
149 if (transpose_a_) { in Compute()
172 transpose_a_, transpose_b_, transpose_c, m, n, k, a_data, offset_a, in Compute()
190 bool transpose_a_; member in tensorflow::QuantizedMatMulOp
|
| D | matmul_op_fused.cc | 152 OP_REQUIRES_OK(context, context->GetAttr("transpose_a", &transpose_a_)); in FusedMatMulOp()
187 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
223 bool transpose_a_; member in tensorflow::FusedMatMulOp
|
| D | sparse_matmul_op.cc | 960 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_a", &transpose_a_)); in SparseMatMulOp()
974 const int m = transpose_a_ ? a.dim_size(1) : a.dim_size(0); in Compute()
975 const int k = transpose_a_ ? a.dim_size(0) : a.dim_size(1); in Compute()
1015 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
1026 bool transpose_a = transpose_a_; in Compute()
1071 bool transpose_a_; member in tensorflow::SparseMatMulOp
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op.cc | 44 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_a", &transpose_a_)); in MklMatMulOp()
58 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
104 bool transpose_a_; member in tensorflow::MklMatMulOp
|
| D | mkl_matmul_op_fused.cc | 36 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_a", &transpose_a_)); in MklFusedMatMulOp()
50 ctx, transpose_a_ == false, in MklFusedMatMulOp()
293 bool transpose_a_; member in tensorflow::MklFusedMatMulOp
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 678 matmul_attrs.transpose_a_ = true; in TEST()
|