| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| runtime_matmul_mkl.cc | 39 int32 transpose_rhs) {
47 int ldb = transpose_rhs ? n : k;
50 transpose_rhs ? CblasTrans : CblasNoTrans, m, n, k, alpha, lhs,
61 int32 transpose_rhs) {
69 int ldb = transpose_rhs ? n : k;
72 transpose_rhs ? CblasTrans : CblasNoTrans, m, n, k, alpha, lhs,
80 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) {
87 MatMulF32(nullptr, out, lhs, rhs, m, n, k, transpose_lhs, transpose_rhs);
95 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) {
102 MatMulF64(nullptr, out, lhs, rhs, m, n, k, transpose_lhs, transpose_rhs);
[all...] |
| runtime_matmul.cc | 41 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) {
53 if (transpose_rhs) {
65 int rhs_contract_dim = transpose_rhs ? 1 : 0;
78 int32 transpose_rhs) {
84 transpose_lhs, transpose_rhs);
90 xla::EigenMatVec<T>(out, lhs, rhs, m, n, k, transpose_lhs, transpose_rhs);
93 transpose_lhs, transpose_rhs);
102 int32 transpose_rhs) {
104 transpose_lhs, transpose_rhs);
109 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) {
[all...] |
| runtime_single_threaded_matmul.cc | 38 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) {
47 if (transpose_rhs) {
59 int rhs_contract_dim = transpose_rhs ? 1 : 0;
72 int32 transpose_lhs, int32 transpose_rhs) {
78 transpose_lhs, transpose_rhs);
82 xla::EigenMatVec<T>(out, lhs, rhs, m, n, k, transpose_lhs, transpose_rhs);
85 transpose_lhs, transpose_rhs);
95 int32 transpose_rhs) {
97 n, k, transpose_lhs, transpose_rhs);
105 int32 transpose_rhs) {
[all...] |
| runtime_matmul.h | 33 tensorflow::int32 transpose_rhs);
39 tensorflow::int32 transpose_rhs);
45 tensorflow::int32 transpose_rhs);
|
| runtime_single_threaded_matmul.h | 33 tensorflow::int32 transpose_rhs);
39 tensorflow::int32 transpose_rhs);
45 tensorflow::int32 transpose_rhs);
|
| 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) {
59 tensorflow::int32 transpose_rhs) {
68 tensorflow::int32 transpose_rhs) {
77 tensorflow::int32 transpose_rhs) {
|
| cpu_runtime_test.cc | 80 bool transpose_rhs,
92 auto b_transpose = MaybeTransposeArray2D(b, !transpose_rhs);
100 m, n, k, transpose_lhs, transpose_rhs);
111 m, n, k, transpose_lhs, transpose_rhs);
132 // * transpose_rhs
142 bool transpose_rhs = std::get<2>(info.param); local
147 transpose_rhs ? "Trhs_" : "",
155 bool transpose_rhs = std::get<2>(GetParam()); local
160 auto c = EigenMatrixMultiply(*a, *b, transpose_lhs, transpose_rhs,
179 bool transpose_rhs = std::get<2>(info.param) local
224 bool transpose_rhs = std::get<2>(GetParam()); local
[all...] |
| runtime_matvec.h | 65 int32 transpose_lhs, int32 transpose_rhs) {
91 transpose_mat = !transpose_rhs;
113 tensorflow::int32 transpose_rhs) {
116 transpose_rhs);
|
| dot_op_emitter.cc | 586 // int32 transpose_rhs);
664 bool transpose_rhs = mat_mult_dims.rhs_non_canonical; local
669 std::swap(transpose_lhs, transpose_rhs);
680 b_->getInt32(transpose_rhs)});
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| tridiagonal_solve_op_test.py | 57 transpose_rhs=False,
61 transpose_rhs, conjugate_rhs)
69 transpose_rhs=False,
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
[all...] |
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| linalg_impl.py | 339 transpose_rhs=False,
382 same left-hand sides and K different right-hand sides. If `transpose_rhs`
399 transpose_rhs: If `True`, `rhs` is transposed before solving (has no effect
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,
503 # Rhs provided as a vector, ignoring transpose_rhs
511 if transpose_rhs:
518 return array_ops.matrix_transpose(result) if transpose_rhs else resul
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| dot_operation_test.cc | 756 for (bool transpose_rhs : {false, true}) {
766 if (transpose_rhs) {
797 if (transpose_rhs) {
804 << transpose_rhs << " " << row_major;
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| algebraic_simplifier_test.cc | 4387 bool transpose_lhs, transpose_rhs; local
[all...] |
| hlo_evaluator.cc | [all...] |
