Home
last modified time | relevance | path

Searched refs:transpose_rhs (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_matmul_mkl.cc39 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 …]
Druntime_single_threaded_matmul.cc35 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs) { in MatMul() argument
44 if (transpose_rhs) { in MatMul()
56 int rhs_contract_dim = transpose_rhs ? 1 : 0; in MatMul()
71 tensorflow::int32 transpose_rhs) { in SingleThreadedMatMulDispatch() argument
77 transpose_lhs, transpose_rhs); in SingleThreadedMatMulDispatch()
81 transpose_lhs, transpose_rhs); in SingleThreadedMatMulDispatch()
91 tensorflow::int32 transpose_rhs) { 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
51 if (transpose_rhs) { in MatMul()
63 int rhs_contract_dim = transpose_rhs ? 1 : 0; in MatMul()
78 tensorflow::int32 transpose_rhs) { in MatMulDispatch() argument
84 transpose_lhs, transpose_rhs); in MatMulDispatch()
89 transpose_lhs, transpose_rhs); in MatMulDispatch()
98 tensorflow::int32 transpose_rhs) { 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
108 transpose_rhs); in __xla_cpu_runtime_EigenMatMulF32()
[all …]
Dcpu_runtime_test.cc79 bool transpose_rhs, in EigenMatrixMultiply() argument
91 auto b_transpose = MaybeTransposeArray2D(b, !transpose_rhs); in EigenMatrixMultiply()
99 m, n, k, transpose_lhs, transpose_rhs); in EigenMatrixMultiply()
109 m, n, k, transpose_lhs, transpose_rhs); in EigenMatrixMultiply()
140 bool transpose_rhs = std::get<2>(info.param); in Name() local
145 transpose_rhs ? "Trhs_" : "", in Name()
153 bool transpose_rhs = std::get<2>(GetParam()); in TEST_P() local
158 auto c = EigenMatrixMultiply(*a, *b, transpose_lhs, transpose_rhs, in TEST_P()
177 bool transpose_rhs = std::get<2>(info.param); in Name() local
182 transpose_rhs ? "Trhs_" : "", in Name()
[all …]
Druntime_matmul_mkl.h28 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
Druntime_matmul.h35 tensorflow::int32 transpose_rhs);
41 tensorflow::int32 transpose_rhs);
47 tensorflow::int32 transpose_rhs);
53 tensorflow::int32 transpose_rhs);
59 tensorflow::int32 transpose_rhs);
65 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs);
Druntime_single_threaded_matmul.h35 tensorflow::int32 transpose_rhs);
41 tensorflow::int32 transpose_rhs);
47 tensorflow::int32 transpose_rhs);
54 tensorflow::int32 transpose_rhs);
61 tensorflow::int32 transpose_rhs);
67 tensorflow::int32 transpose_lhs, tensorflow::int32 transpose_rhs);
Ddot_op_emitter.cc830 bool transpose_rhs = !mat_mult_dims.rhs_canonical; in EmitCallToRuntime() local
835 std::swap(transpose_lhs, transpose_rhs); in EmitCallToRuntime()
846 b_->getInt32(transpose_rhs)}); in EmitCallToRuntime()
/external/tensorflow/tensorflow/python/kernel_tests/
Dtridiagonal_solve_op_test.py78 transpose_rhs=False, argument
91 transpose_rhs,
101 transpose_rhs=False, argument
105 transpose_rhs, conjugate_rhs)
336 transpose_rhs=True)
352 transpose_rhs=True,
363 transpose_rhs=True)
370 transpose_rhs=True)
384 transpose_rhs=True)
396 transpose_rhs=False, argument
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dtridiagonal_solve_ops_test.py146 transpose_rhs=False): argument
154 transpose_rhs,
164 transpose_rhs=False): argument
171 transpose_rhs,
357 transpose_rhs=True)
368 transpose_rhs=True)
378 transpose_rhs=True)
390 transpose_rhs=False, argument
403 transpose_rhs=transpose_rhs,
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinalg_impl.py451 transpose_rhs=False, argument
547 return _tridiagonal_solve_compact_format(diagonals, rhs, transpose_rhs,
580 return _tridiagonal_solve_compact_format(diagonals, rhs, transpose_rhs,
595 diagonals, rhs, transpose_rhs, conjugate_rhs, partial_pivoting, name)
600 def _tridiagonal_solve_compact_format(diagonals, rhs, transpose_rhs, argument
639 if transpose_rhs:
Dlinear_operator_tridiag.py339 transpose_rhs=adjoint_arg,
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddot_operation_test.cc796 for (bool transpose_rhs : {false, true}) { in XLA_TYPED_TEST()
806 if (transpose_rhs) { in XLA_TYPED_TEST()
837 if (transpose_rhs) { in XLA_TYPED_TEST()
844 << transpose_rhs << " " << row_major; in XLA_TYPED_TEST()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.linalg.pbtxt265 …argspec: "args=[\'diagonals\', \'rhs\', \'diagonals_format\', \'transpose_rhs\', \'conjugate_rhs\'…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.linalg.pbtxt273 …argspec: "args=[\'diagonals\', \'rhs\', \'diagonals_format\', \'transpose_rhs\', \'conjugate_rhs\'…
/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()
5262 1, transpose_rhs ? transposed_rhs_shape : rhs_shape, "rhs")); in TEST_P()
5263 if (transpose_rhs) { 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()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc540 bool transpose_rhs, int64_t num_batch_dims, in BatchDot() argument
548 llvm::makeArrayRef({transpose_rhs ? num_batch_dims + 1 : num_batch_dims}), in BatchDot()
562 transpose_rhs ? rhs_shape[rhs_shape.size() - 2] : rhs_shape.back(); in BatchDot()