Searched refs:rhs_transpose (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 63 const bool rhs_transpose = (rhs_contracting_dim_number - num_batch_dims) == 1; in DecomposeBatchDot() local
127 if (rhs_transpose) { in DecomposeBatchDot()
240 std::vector<int64> rhs_transpose; in CanonicalizeDot() local
241 rhs_transpose.reserve(rhs_rank); in CanonicalizeDot()
242 rhs_transpose.insert(rhs_transpose.end(), in CanonicalizeDot()
245 rhs_transpose.insert(rhs_transpose.end(), in CanonicalizeDot()
248 rhs_transpose.insert(rhs_transpose.end(), rhs_non_contracting_dims.begin(), in CanonicalizeDot()
252 ShapeUtil::PermuteDimensions(InversePermutation(rhs_transpose), in CanonicalizeDot()
254 original_dot->mutable_operand(1), rhs_transpose)); in CanonicalizeDot()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 68 auto rhs_transpose = rhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemm() local
75 lhs_transpose, rhs_transpose, output_matrix.num_rows, in DoGemm()
88 lhs_transpose, rhs_transpose, output_matrix.num_rows, in DoGemm()
133 auto rhs_transpose = rhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemmWithAlgorithm() local
139 lhs_transpose, rhs_transpose, output_matrix.num_rows, in DoGemmWithAlgorithm()
|