| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_matmul_mkl.cc | 39 int32 transpose_rhs) { in MatMulF32() argument
47 int ldb = transpose_rhs ? n : k; in MatMulF32()
50 transpose_rhs ? CblasTrans : CblasNoTrans, m, n, k, alpha, lhs, in MatMulF32()
61 int32 transpose_rhs) { in MatMulF64() argument
69 int ldb = transpose_rhs ? n : k; in MatMulF64()
72 transpose_rhs ? CblasTrans : CblasNoTrans, m, n, k, alpha, lhs, 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
47 if (transpose_rhs) { in MatMul()
59 int rhs_contract_dim = transpose_rhs ? 1 : 0; 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()
95 int32 transpose_rhs) { in __xla_cpu_runtime_EigenSingleThreadedMatMulF16() argument
97 n, k, transpose_lhs, transpose_rhs); in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
105 int32 transpose_rhs) { 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
53 if (transpose_rhs) { in MatMul()
65 int rhs_contract_dim = transpose_rhs ? 1 : 0; in MatMul()
78 int32 transpose_rhs) { 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()
102 int32 transpose_rhs) { 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 | 80 bool transpose_rhs, in EigenMatrixMultiply() argument
92 auto b_transpose = MaybeTransposeArray2D(b, !transpose_rhs); in EigenMatrixMultiply()
100 m, n, k, transpose_lhs, transpose_rhs); in EigenMatrixMultiply()
111 m, n, k, transpose_lhs, transpose_rhs); in EigenMatrixMultiply()
142 bool transpose_rhs = std::get<2>(info.param); in Name() local
147 transpose_rhs ? "Trhs_" : "", in Name()
155 bool transpose_rhs = std::get<2>(GetParam()); in TEST_P() local
160 auto c = EigenMatrixMultiply(*a, *b, transpose_lhs, transpose_rhs, in TEST_P()
179 bool transpose_rhs = std::get<2>(info.param); in Name() local
184 transpose_rhs ? "Trhs_" : "", in Name()
[all …]
|
| D | runtime_matmul_mkl.h | 28 tensorflow::int32 transpose_rhs);
33 tensorflow::int32 transpose_rhs);
38 tensorflow::int32 transpose_rhs);
43 tensorflow::int32 transpose_rhs);
50 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF32() argument
59 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_MKLMatMulF64() argument
68 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF32() argument
77 tensorflow::int32 transpose_rhs) { in __xla_cpu_runtime_MKLSingleThreadedMatMulF64() argument
|
| D | runtime_matvec.h | 65 int32 transpose_lhs, int32 transpose_rhs) { in DispatchMatVec() argument
91 transpose_mat = !transpose_rhs; in DispatchMatVec()
113 tensorflow::int32 transpose_rhs) { in EigenMatVec() argument
116 transpose_rhs); in EigenMatVec()
|
| D | runtime_matmul.h | 33 tensorflow::int32 transpose_rhs);
39 tensorflow::int32 transpose_rhs);
45 tensorflow::int32 transpose_rhs);
|
| D | runtime_single_threaded_matmul.h | 33 tensorflow::int32 transpose_rhs);
39 tensorflow::int32 transpose_rhs);
45 tensorflow::int32 transpose_rhs);
|
| D | dot_op_emitter.cc | 664 bool transpose_rhs = mat_mult_dims.rhs_non_canonical; in EmitCallToRuntime() local
669 std::swap(transpose_lhs, transpose_rhs); in EmitCallToRuntime()
680 b_->getInt32(transpose_rhs)}); in EmitCallToRuntime()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tridiagonal_solve_op_test.py | 57 transpose_rhs=False, argument
61 transpose_rhs, conjugate_rhs)
69 transpose_rhs=False, argument
73 transpose_rhs, conjugate_rhs)
244 transpose_rhs=True)
260 transpose_rhs=True,
270 transpose_rhs=True)
277 transpose_rhs=True)
291 transpose_rhs=True)
444 x = linalg_impl.tridiagonal_solve(diags, rhs, transpose_rhs=True)
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linalg_impl.py | 339 transpose_rhs=False, argument
413 return _tridiagonal_solve_compact_format(diagonals, rhs, transpose_rhs,
445 return _tridiagonal_solve_compact_format(diagonals, rhs, transpose_rhs,
470 return _tridiagonal_solve_compact_format(diagonals, rhs, transpose_rhs,
478 transpose_rhs=False, argument
511 if transpose_rhs:
518 return array_ops.matrix_transpose(result) if transpose_rhs else result
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 756 for (bool transpose_rhs : {false, true}) { in XLA_TYPED_TEST()
766 if (transpose_rhs) { in XLA_TYPED_TEST()
797 if (transpose_rhs) { in XLA_TYPED_TEST()
804 << transpose_rhs << " " << row_major; in XLA_TYPED_TEST()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.linalg.pbtxt | 209 …argspec: "args=[\'diagonals\', \'rhs\', \'diagonals_format\', \'transpose_rhs\', \'conjugate_rhs\'…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.linalg.pbtxt | 209 …argspec: "args=[\'diagonals\', \'rhs\', \'diagonals_format\', \'transpose_rhs\', \'conjugate_rhs\'…
|
| /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()
4405 1, transpose_rhs ? transposed_rhs_shape : rhs_shape, "rhs")); in TEST_P()
4406 if (transpose_rhs) { 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()
|