| /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 | 38 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in MatMul() argument
41 if (transpose_lhs) { in MatMul()
58 int lhs_contract_dim = transpose_lhs ? 0 : 1; in MatMul()
72 int32 transpose_lhs, int32 transpose_rhs) { in SingleThreadedMatMulDispatch() argument
78 transpose_lhs, transpose_rhs); in SingleThreadedMatMulDispatch()
82 xla::EigenMatVec<T>(out, lhs, rhs, m, n, k, transpose_lhs, transpose_rhs); in SingleThreadedMatMulDispatch()
85 transpose_lhs, transpose_rhs); in SingleThreadedMatMulDispatch()
94 Eigen::half* rhs, int64 m, int64 n, int64 k, int32 transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16() argument
97 n, k, transpose_lhs, transpose_rhs); in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
104 int64 k, int32 transpose_lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF32() argument
[all …]
|
| D | runtime_matmul.cc | 41 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in MatMul() argument
47 if (transpose_lhs) { in MatMul()
64 int lhs_contract_dim = transpose_lhs ? 0 : 1; in MatMul()
77 int64 m, int64 n, int64 k, int32 transpose_lhs, in MatMulDispatch() argument
84 transpose_lhs, transpose_rhs); in MatMulDispatch()
90 xla::EigenMatVec<T>(out, lhs, rhs, m, n, k, transpose_lhs, transpose_rhs); in MatMulDispatch()
93 transpose_lhs, transpose_rhs); in MatMulDispatch()
101 Eigen::half* rhs, int64 m, int64 n, int64 k, int32 transpose_lhs, in __xla_cpu_runtime_EigenMatMulF16() argument
104 transpose_lhs, transpose_rhs); in __xla_cpu_runtime_EigenMatMulF16()
109 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) { in __xla_cpu_runtime_EigenMatMulF32() argument
[all …]
|
| D | cpu_runtime_test.cc | 79 bool transpose_lhs, in EigenMatrixMultiply() argument
91 auto a_transpose = MaybeTransposeArray2D(a, !transpose_lhs); in EigenMatrixMultiply()
100 m, n, k, transpose_lhs, transpose_rhs); in EigenMatrixMultiply()
111 m, n, k, transpose_lhs, transpose_rhs); in EigenMatrixMultiply()
141 bool transpose_lhs = std::get<1>(info.param); in Name() local
146 shape.k, shape.n, transpose_lhs ? "Tlhs_" : "", in Name()
154 bool transpose_lhs = std::get<1>(GetParam()); in TEST_P() local
160 auto c = EigenMatrixMultiply(*a, *b, transpose_lhs, transpose_rhs, in TEST_P()
178 bool transpose_lhs = std::get<1>(info.param); in Name() local
183 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_matvec.h | 65 int32 transpose_lhs, int32 transpose_rhs) { in DispatchMatVec() argument
86 transpose_mat = transpose_lhs; in DispatchMatVec()
112 tensorflow::int32 transpose_lhs, in EigenMatVec() argument
115 detail::DispatchMatVec<T>(out, lhs, rhs, m, n, k, transpose_lhs, in EigenMatVec()
|
| D | runtime_matmul.h | 32 tensorflow::int64 n, tensorflow::int64 k, tensorflow::int32 transpose_lhs,
38 tensorflow::int64 k, tensorflow::int32 transpose_lhs,
44 tensorflow::int64 k, tensorflow::int32 transpose_lhs,
|
| D | runtime_single_threaded_matmul.h | 32 tensorflow::int64 n, tensorflow::int64 k, tensorflow::int32 transpose_lhs,
38 tensorflow::int64 k, tensorflow::int32 transpose_lhs,
44 tensorflow::int64 k, tensorflow::int32 transpose_lhs,
|
| D | dot_op_emitter.cc | 663 bool transpose_lhs = mat_mult_dims.lhs_non_canonical; in EmitCallToRuntime() local
669 std::swap(transpose_lhs, transpose_rhs); in EmitCallToRuntime()
679 b_->getInt64(mat_mult_dims.k), b_->getInt32(transpose_lhs), in EmitCallToRuntime()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 755 for (bool transpose_lhs : {false, true}) { in XLA_TYPED_TEST()
763 if (transpose_lhs) { in XLA_TYPED_TEST()
794 if (transpose_lhs) { in XLA_TYPED_TEST()
803 VLOG(1) << "TestTransposeFolding " << transpose_lhs << " " in XLA_TYPED_TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 4387 bool transpose_lhs, transpose_rhs; in TEST_P() local
4389 std::tie(m, k, n, transpose_lhs, transpose_rhs, element_type) = GetParam(); in TEST_P()
4399 0, transpose_lhs ? transposed_lhs_shape : lhs_shape, "lhs")); in TEST_P()
4400 if (transpose_lhs) { in TEST_P()
4420 dot_should_be_transformed || (transpose_lhs && transpose_rhs); in TEST_P()
|
| D | hlo_evaluator.cc | 1620 int64 k, int32 transpose_lhs, int32 transpose_rhs)>& impl_fn) { in MatmulArray2DImpl() argument
|