| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_runtime_test.cc | 79 bool transpose_lhs,
91 auto a_transpose = MaybeTransposeArray2D(a, !transpose_lhs);
100 m, n, k, transpose_lhs, transpose_rhs);
111 m, n, k, transpose_lhs, transpose_rhs);
131 // * transpose_lhs
141 bool transpose_lhs = std::get<1>(info.param); local
146 shape.k, shape.n, transpose_lhs ? "Tlhs_" : "",
154 bool transpose_lhs = std::get<1>(GetParam()); local
160 auto c = EigenMatrixMultiply(*a, *b, transpose_lhs, transpose_rhs,
178 bool transpose_lhs = std::get<1>(info.param) local
223 bool transpose_lhs = std::get<1>(GetParam()); local
[all...] |
| dot_op_emitter.cc | 585 // int64 m, int64 n, int64 k, int32 transpose_lhs,
663 bool transpose_lhs = mat_mult_dims.lhs_non_canonical; local
669 std::swap(transpose_lhs, transpose_rhs);
679 b_->getInt64(mat_mult_dims.k), b_->getInt32(transpose_lhs),
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| algebraic_simplifier_test.cc | 4387 bool transpose_lhs, transpose_rhs; local
[all...] |
