| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | triangular_solve_thunk.cc | 55 transpose_a_ = [&] { in TriangularSolveThunk()
98 ->ThenBlasTrsm(side_, uplo_, transpose_a_, in ExecuteOnStream()
108 ->ThenBlasTrsm(side_, uplo_, transpose_a_, in ExecuteOnStream()
119 ->ThenBlasTrsm(side_, uplo_, transpose_a_, unit_diagonal_, m_, in ExecuteOnStream()
130 ->ThenBlasTrsm(side_, uplo_, transpose_a_, unit_diagonal_, m_, in ExecuteOnStream()
|
| D | triangular_solve_thunk.h | 59 se::blas::Transpose transpose_a_; variable
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matmul_op.cc | 34 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_a", &transpose_a_)); in MatMulOp()
65 int first_index = transpose_a_ ? 0 : 1; in Compile()
84 auto lhs = (transpose_a_) ? xla::Transpose(a, {1, 0}) : a; in Compile()
91 bool transpose_a_; member in tensorflow::__anondc94a7a40111::MatMulOp
|
| /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 | mkl_matmul_op.cc | 48 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_a", &transpose_a_)); in MklMatMulOp()
62 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
107 bool transpose_a_; member in tensorflow::MklMatMulOp
|
| D | matmul_op.cc | 443 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_a", &transpose_a_)); in MatMulOp()
465 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
526 bool transpose_a_; member in tensorflow::MatMulOp
|
| D | sparse_matmul_op.cc | 961 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_a", &transpose_a_)); in SparseMatMulOp()
975 const int m = transpose_a_ ? a.dim_size(1) : a.dim_size(0); in Compute()
976 const int k = transpose_a_ ? a.dim_size(0) : a.dim_size(1); in Compute()
1016 dim_pair[0].first = transpose_a_ ? 0 : 1; in Compute()
1027 bool transpose_a = transpose_a_; in Compute()
1072 bool transpose_a_; member in tensorflow::SparseMatMulOp
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 73 matmul_attrs.transpose_a_ = true; in TEST()
|