Home
last modified time | relevance | path

Searched refs:rhs_transpose (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_decomposer.cc117 std::vector<int64> rhs_transpose; in CanonicalizeDot() local
118 rhs_transpose.reserve(rhs_rank); in CanonicalizeDot()
119 rhs_transpose.insert(rhs_transpose.end(), in CanonicalizeDot()
122 rhs_transpose.insert(rhs_transpose.end(), in CanonicalizeDot()
125 rhs_transpose.insert(rhs_transpose.end(), rhs_non_contracting_dims.begin(), in CanonicalizeDot()
129 ShapeUtil::PermuteDimensions(rhs_transpose, rhs_shape), in CanonicalizeDot()
130 original_dot->mutable_operand(1), rhs_transpose)); in CanonicalizeDot()
Dalgebraic_simplifier.cc2282 HloInstruction* rhs_transpose = in OptimizeDotOfReorderContractingDims() local
2287 rhs = rhs_transpose; in OptimizeDotOfReorderContractingDims()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Deinsum.cc190 std::vector<int32_t> rhs_transpose; in transposeForBatchMatmul() local
194 rhs_transpose.reserve(dnums.lhs_rhs_out.size() + dnums.rhs_out.size() + in transposeForBatchMatmul()
200 rhs_transpose.push_back(std::get<1>(dnums.lhs_rhs_out[i])); in transposeForBatchMatmul()
213 rhs_transpose.push_back(std::get<1>(dnums.lhs_rhs[i])); in transposeForBatchMatmul()
215 std::make_tuple(lhs_transpose.size() - 1, rhs_transpose.size() - 1); in transposeForBatchMatmul()
218 rhs_transpose.push_back(std::get<0>(dnums.rhs_out[i])); in transposeForBatchMatmul()
221 std::make_tuple(rhs_transpose.size() - 1, out_transpose.size() - 1); in transposeForBatchMatmul()
230 *rhs = createTransposeOp(*rhs, loc, rhs_transpose, rewriter); in transposeForBatchMatmul()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc123 auto rhs_transpose = rhs_matrix.transpose ? se::blas::Transpose::kTranspose in DoGemmWithAlgorithm() local
132 lhs_transpose, rhs_transpose, output_matrix.num_rows, in DoGemmWithAlgorithm()
150 lhs_transpose, rhs_transpose, output_matrix.num_rows, in DoGemmWithAlgorithm()
163 lhs_transpose, rhs_transpose, output_matrix.num_rows, in DoGemmWithAlgorithm()