Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_matmul.cc37 int64_t n, int64_t k, int32_t transpose_lhs, in MatMul() argument
44 if (transpose_lhs) { in MatMul()
61 int lhs_contract_dim = transpose_lhs ? 0 : 1; in MatMul()
76 int32_t transpose_lhs, int32_t transpose_rhs) { in MatMul_Batch() argument
82 if (transpose_lhs) { in MatMul_Batch()
99 int lhs_contract_dim = transpose_lhs ? 0 : 1; in MatMul_Batch()
118 int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in MatMulDispatch() argument
125 transpose_lhs, transpose_rhs); in MatMulDispatch()
130 transpose_lhs, transpose_rhs); in MatMulDispatch()
136 int32_t transpose_lhs, int32_t transpose_rhs) { in BatchMatMulDispatch() argument
[all …]
Druntime_single_threaded_matmul.cc33 int64_t n, int64_t k, int32_t transpose_lhs, in MatMul() argument
37 if (transpose_lhs) { in MatMul()
54 int lhs_contract_dim = transpose_lhs ? 0 : 1; in MatMul()
68 int32_t transpose_lhs, in SingleThreadedMatMulDispatch() argument
75 transpose_lhs, transpose_rhs); in SingleThreadedMatMulDispatch()
79 transpose_lhs, transpose_rhs); in SingleThreadedMatMulDispatch()
87 Eigen::half* rhs, int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16() argument
90 n, k, transpose_lhs, transpose_rhs); in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
97 int64_t k, int32_t transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF32() argument
100 transpose_lhs, transpose_rhs); in __xla_cpu_runtime_EigenSingleThreadedMatMulF32()
[all …]
Druntime_matmul_mkl.cc34 int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in MatMulF32() argument
42 int lda = transpose_lhs ? k : m; in MatMulF32()
45 cblas_sgemm(CblasColMajor, transpose_lhs ? CblasTrans : CblasNoTrans, in MatMulF32()
57 int32_t transpose_lhs, int32_t transpose_rhs) { in MatMulF64() argument
64 int lda = transpose_lhs ? k : m; in MatMulF64()
67 cblas_dgemm(CblasColMajor, transpose_lhs ? CblasTrans : CblasNoTrans, in MatMulF64()
76 int64_t n, int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF32() argument
83 MatMulF32(nullptr, out, lhs, rhs, m, n, k, transpose_lhs, transpose_rhs); in __xla_cpu_runtime_MKLMatMulF32()
91 int64_t m, int64_t n, int64_t k, int32_t transpose_lhs, in __xla_cpu_runtime_MKLMatMulF64() argument
99 MatMulF64(nullptr, out, lhs, rhs, m, n, k, transpose_lhs, transpose_rhs); in __xla_cpu_runtime_MKLMatMulF64()
[all …]
Dcpu_runtime_test.cc80 bool transpose_lhs, in EigenMatrixMultiply() argument
92 auto a_transpose = MaybeTransposeArray2D(a, !transpose_lhs); in EigenMatrixMultiply()
101 m, n, k, transpose_lhs, transpose_rhs); in EigenMatrixMultiply()
111 m, n, k, transpose_lhs, transpose_rhs); in EigenMatrixMultiply()
142 bool transpose_lhs = std::get<1>(info.param); in Name() local
147 shape.k, shape.n, transpose_lhs ? "Tlhs_" : "", in Name()
155 bool transpose_lhs = std::get<1>(GetParam()); in TEST_P() local
161 auto c = EigenMatrixMultiply(*a, *b, transpose_lhs, transpose_rhs, in TEST_P()
180 bool transpose_lhs = std::get<1>(info.param); in Name() local
185 shape.k, shape.n, transpose_lhs ? "Tlhs_" : "", in Name()
[all …]
Druntime_matmul_mkl.h27 int32_t transpose_lhs, int32_t transpose_rhs);
31 int32_t transpose_lhs, int32_t transpose_rhs);
35 int32_t transpose_lhs, int32_t transpose_rhs);
39 int32_t transpose_lhs, int32_t transpose_rhs);
45 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF32() argument
53 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF64() argument
61 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF32() argument
69 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF64() argument
Druntime_matmul.h35 int64_t k, int32_t transpose_lhs, int32_t transpose_rhs);
40 int32_t transpose_lhs, int32_t transpose_rhs);
45 int32_t transpose_lhs, int32_t transpose_rhs);
50 int64_t k, int32_t transpose_lhs, int32_t transpose_rhs);
55 int64_t k, int32_t transpose_lhs, int32_t transpose_rhs);
60 int32_t transpose_lhs, int32_t transpose_rhs);
65 int32_t transpose_lhs, int32_t transpose_rhs);
Druntime_matmul_acl.cc38 int64_t batch_size, int32_t transpose_lhs, in MatMulF32() argument
49 acl_conf.is_trans_lhs = (bool)transpose_lhs; in MatMulF32()
145 if (transpose_lhs && !transpose_rhs) { in MatMulF32()
150 } else if (transpose_rhs && !transpose_lhs) { in MatMulF32()
155 } else if (transpose_rhs && transpose_lhs) { in MatMulF32()
185 int64_t n, int64_t k, int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_ACLMatMulF32() argument
187 transpose_lhs, transpose_rhs) < 0) { in __xla_cpu_runtime_ACLMatMulF32()
190 transpose_lhs, transpose_rhs); in __xla_cpu_runtime_ACLMatMulF32()
196 int64_t n, int64_t k, int64_t batch_size, int32_t transpose_lhs, in __xla_cpu_runtime_ACLBatchMatMulF32() argument
199 transpose_lhs, transpose_rhs) < 0) { in __xla_cpu_runtime_ACLBatchMatMulF32()
[all …]
Druntime_single_threaded_matmul.h33 int64_t k, int32_t transpose_lhs, int32_t transpose_rhs);
38 int32_t transpose_lhs, int32_t transpose_rhs);
43 int32_t transpose_lhs, int32_t transpose_rhs);
49 int32_t transpose_lhs, int32_t transpose_rhs);
55 int32_t transpose_lhs, int32_t transpose_rhs);
60 int32_t transpose_lhs, int32_t transpose_rhs);
Druntime_matmul_acl.h56 int32_t transpose_lhs, int32_t transpose_rhs);
61 int32_t transpose_lhs, int32_t transpose_rhs);
69 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_ACLMatMulF32() argument
79 int32_t transpose_lhs, int32_t transpose_rhs) { in __xla_cpu_runtime_ACLBatchMatMulF32() argument
Ddot_op_emitter.cc896 bool transpose_lhs = !mat_mult_dims.lhs_canonical; in EmitCallToRuntime() local
902 std::swap(transpose_lhs, transpose_rhs); in EmitCallToRuntime()
912 b_->getInt64(mat_mult_dims.k), b_->getInt32(transpose_lhs), in EmitCallToRuntime()
978 bool transpose_lhs = !mat_mult_dims.lhs_canonical; in EmitCallToBatchRuntime() local
985 std::swap(transpose_lhs, transpose_rhs); in EmitCallToBatchRuntime()
998 b_->getInt32(static_cast<uint32_t>(transpose_lhs)), in EmitCallToBatchRuntime()
/external/ComputeLibrary/examples/gemm_tuner/
Dcl_gemmlowp_reshaped.cpp62 bool transpose_lhs{ true }; /**< Transpose lhs matrix. */ member
85 os << "transpose_lhs : " << (configs.transpose_lhs ? true_str : false_str) << std::endl; in operator <<()
163 configs.transpose_lhs = options.transpose_rhs->value() == 0; in consume_gemm_configs()
233 lhs_info.transpose = configs.transpose_lhs; in do_setup()
Dcl_gemm_reshaped.cpp62 bool transpose_lhs{ true }; /**< Transpose lhs matrix. */ member
86 os << "transpose_lhs : " << (configs.transpose_lhs ? true_str : false_str) << std::endl; in operator <<()
168 configs.transpose_lhs = options.transpose_rhs->value() == 0; in consume_gemm_configs()
240 lhs_info.transpose = configs.transpose_lhs; in do_setup()
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddot_operation_test.cc796 for (bool transpose_lhs : {false, true}) { in XLA_TYPED_TEST()
804 if (transpose_lhs) { in XLA_TYPED_TEST()
835 if (transpose_lhs) { in XLA_TYPED_TEST()
844 VLOG(1) << "TestTransposeFolding " << transpose_lhs << " " in XLA_TYPED_TEST()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tfl.cc1361 bool transpose_lhs = tfl_mm_op.adj_x(); in matchAndRewrite() local
1404 if (transpose_lhs) { in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc5885 bool transpose_lhs, transpose_rhs; in TEST_P() local
5887 std::tie(m, k, n, transpose_lhs, transpose_rhs, element_type) = GetParam(); in TEST_P()
5897 0, transpose_lhs ? transposed_lhs_shape : lhs_shape, "lhs")); in TEST_P()
5898 if (transpose_lhs) { in TEST_P()
5920 dot_should_be_transformed || (transpose_lhs && transpose_rhs); in TEST_P()
Dhlo_evaluator.cc3882 int32_t transpose_lhs, int32_t transpose_rhs)>& in MatmulArray2DImpl() argument
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc441 Value BatchDot(Location loc, Value lhs, bool transpose_lhs, Value rhs, in BatchDot() argument
447 {transpose_lhs ? num_batch_dims : num_batch_dims + 1})); in BatchDot()
460 transpose_lhs ? lhs_shape.back() : lhs_shape[lhs_shape.size() - 2]; in BatchDot()