Home
last modified time | relevance | path

Searched refs:transpose_lhs (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_matmul_mkl.cc38 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 …]
Druntime_single_threaded_matmul.cc35 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 …]
Druntime_matmul.cc39 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 …]
Dcpu_runtime_test.cc78 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 …]
Druntime_matmul_mkl.h27 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
Druntime_matmul.h34 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);
Druntime_single_threaded_matmul.h34 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);
Ddot_op_emitter.cc829 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/
Ddot_operation_test.cc795 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/
Dalgebraic_simplifier_test.cc5244 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()
Dhlo_evaluator.cc2560 int64 k, int32 transpose_lhs, int32 transpose_rhs)>& impl_fn) { in MatmulArray2DImpl() argument
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc539 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()