/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | runtime_matmul_mkl.cc | 38 int64 m, int64 n, int64 k, int32 transpose_lhs, in MatMulF32() argument 46 int lda = transpose_lhs ? k : m; in MatMulF32() 49 cblas_sgemm(CblasColMajor, transpose_lhs ? CblasTrans : CblasNoTrans, in MatMulF32() 60 double* rhs, int64 m, int64 n, int64 k, int32 transpose_lhs, in MatMulF64() argument 68 int lda = transpose_lhs ? k : m; in MatMulF64() 71 cblas_dgemm(CblasColMajor, transpose_lhs ? CblasTrans : CblasNoTrans, in MatMulF64() 80 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF32() argument 87 MatMulF32(nullptr, out, lhs, rhs, m, n, k, transpose_lhs, transpose_rhs); in __xla_cpu_runtime_MKLMatMulF32() 95 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF64() argument 102 MatMulF64(nullptr, out, lhs, rhs, m, n, k, transpose_lhs, transpose_rhs); in __xla_cpu_runtime_MKLMatMulF64() [all …]
|
D | runtime_single_threaded_matmul.cc | 35 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in MatMul() argument 38 if (transpose_lhs) { in MatMul() 55 int lhs_contract_dim = transpose_lhs ? 0 : 1; in MatMul() 70 tensorflow::int32 transpose_lhs, in SingleThreadedMatMulDispatch() argument 77 transpose_lhs, transpose_rhs); in SingleThreadedMatMulDispatch() 81 transpose_lhs, transpose_rhs); in SingleThreadedMatMulDispatch() 90 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16() argument 93 n, k, transpose_lhs, transpose_rhs); in __xla_cpu_runtime_EigenSingleThreadedMatMulF16() 100 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulF32() argument 102 transpose_lhs, transpose_rhs); in __xla_cpu_runtime_EigenSingleThreadedMatMulF32() [all …]
|
D | runtime_matmul.cc | 39 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in MatMul() argument 45 if (transpose_lhs) { in MatMul() 62 int lhs_contract_dim = transpose_lhs ? 0 : 1; in MatMul() 77 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in MatMulDispatch() argument 84 transpose_lhs, transpose_rhs); in MatMulDispatch() 89 transpose_lhs, transpose_rhs); in MatMulDispatch() 97 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_EigenMatMulF16() argument 100 transpose_lhs, transpose_rhs); in __xla_cpu_runtime_EigenMatMulF16() 106 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulF32() argument 107 MatMulDispatch<float>(run_options_ptr, out, lhs, rhs, m, n, k, transpose_lhs, in __xla_cpu_runtime_EigenMatMulF32() [all …]
|
D | cpu_runtime_test.cc | 78 bool transpose_lhs, in EigenMatrixMultiply() argument 90 auto a_transpose = MaybeTransposeArray2D(a, !transpose_lhs); in EigenMatrixMultiply() 99 m, n, k, transpose_lhs, transpose_rhs); in EigenMatrixMultiply() 109 m, n, k, transpose_lhs, transpose_rhs); in EigenMatrixMultiply() 139 bool transpose_lhs = std::get<1>(info.param); in Name() local 144 shape.k, shape.n, transpose_lhs ? "Tlhs_" : "", in Name() 152 bool transpose_lhs = std::get<1>(GetParam()); in TEST_P() local 158 auto c = EigenMatrixMultiply(*a, *b, transpose_lhs, transpose_rhs, in TEST_P() 176 bool transpose_lhs = std::get<1>(info.param); in Name() local 181 shape.k, shape.n, transpose_lhs ? "Tlhs_" : "", in Name() [all …]
|
D | runtime_matmul_mkl.h | 27 tensorflow::int64 k, tensorflow::int32 transpose_lhs, 32 tensorflow::int64 k, tensorflow::int32 transpose_lhs, 37 tensorflow::int64 k, tensorflow::int32 transpose_lhs, 42 tensorflow::int64 k, tensorflow::int32 transpose_lhs, 49 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_MKLMatMulF32() argument 58 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_MKLMatMulF64() argument 67 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_MKLSingleThreadedMatMulF32() argument 76 tensorflow::int64 k, tensorflow::int32 transpose_lhs, in __xla_cpu_runtime_MKLSingleThreadedMatMulF64() argument
|
D | runtime_matmul.h | 34 tensorflow::int64 n, tensorflow::int64 k, tensorflow::int32 transpose_lhs, 40 tensorflow::int64 k, tensorflow::int32 transpose_lhs, 46 tensorflow::int64 k, tensorflow::int32 transpose_lhs, 52 tensorflow::int64 n, tensorflow::int64 k, tensorflow::int32 transpose_lhs, 58 tensorflow::int64 n, tensorflow::int64 k, tensorflow::int32 transpose_lhs, 65 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs);
|
D | runtime_single_threaded_matmul.h | 34 tensorflow::int64 n, tensorflow::int64 k, tensorflow::int32 transpose_lhs, 40 tensorflow::int64 k, tensorflow::int32 transpose_lhs, 46 tensorflow::int64 k, tensorflow::int32 transpose_lhs, 53 tensorflow::int64 k, tensorflow::int32 transpose_lhs, 60 tensorflow::int64 k, tensorflow::int32 transpose_lhs, 67 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs);
|
D | dot_op_emitter.cc | 829 bool transpose_lhs = !mat_mult_dims.lhs_canonical; in EmitCallToRuntime() local 835 std::swap(transpose_lhs, transpose_rhs); in EmitCallToRuntime() 845 b_->getInt64(mat_mult_dims.k), b_->getInt32(transpose_lhs), in EmitCallToRuntime()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | dot_operation_test.cc | 795 for (bool transpose_lhs : {false, true}) { in XLA_TYPED_TEST() 803 if (transpose_lhs) { in XLA_TYPED_TEST() 834 if (transpose_lhs) { in XLA_TYPED_TEST() 843 VLOG(1) << "TestTransposeFolding " << transpose_lhs << " " in XLA_TYPED_TEST()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | algebraic_simplifier_test.cc | 5244 bool transpose_lhs, transpose_rhs; in TEST_P() local 5246 std::tie(m, k, n, transpose_lhs, transpose_rhs, element_type) = GetParam(); in TEST_P() 5256 0, transpose_lhs ? transposed_lhs_shape : lhs_shape, "lhs")); in TEST_P() 5257 if (transpose_lhs) { in TEST_P() 5279 dot_should_be_transformed || (transpose_lhs && transpose_rhs); in TEST_P()
|
D | hlo_evaluator.cc | 2560 int64 k, int32 transpose_lhs, int32 transpose_rhs)>& impl_fn) { in MatmulArray2DImpl() argument
|
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
D | legalize_tf.cc | 539 Value BatchDot(Location loc, Value lhs, bool transpose_lhs, Value rhs, in BatchDot() argument 545 llvm::makeArrayRef({transpose_lhs ? num_batch_dims : num_batch_dims + 1}), in BatchDot() 560 transpose_lhs ? lhs_shape.back() : lhs_shape[lhs_shape.size() - 2]; in BatchDot()
|