Searched refs:lhs_transpose (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 71 std::vector<int64> lhs_transpose; in CanonicalizeDot() local
72 lhs_transpose.reserve(lhs_rank); in CanonicalizeDot()
73 lhs_transpose.insert(lhs_transpose.end(), in CanonicalizeDot()
76 lhs_transpose.insert(lhs_transpose.end(), lhs_non_contracting_dims.begin(), in CanonicalizeDot()
78 lhs_transpose.insert(lhs_transpose.end(), in CanonicalizeDot()
83 ShapeUtil::PermuteDimensions(lhs_transpose, lhs_shape), in CanonicalizeDot()
84 original_dot->mutable_operand(0), lhs_transpose)); in CanonicalizeDot()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | einsum.cc | 189 std::vector<int32_t> lhs_transpose; in transposeForBatchMatmul() local
192 lhs_transpose.reserve(dnums.lhs_rhs_out.size() + dnums.lhs_out.size() + in transposeForBatchMatmul()
199 lhs_transpose.push_back(std::get<0>(dnums.lhs_rhs_out[i])); in transposeForBatchMatmul()
206 lhs_transpose.push_back(std::get<0>(dnums.lhs_out[i])); in transposeForBatchMatmul()
209 std::make_tuple(lhs_transpose.size() - 1, out_transpose.size() - 1); in transposeForBatchMatmul()
212 lhs_transpose.push_back(std::get<0>(dnums.lhs_rhs[i])); in transposeForBatchMatmul()
215 std::make_tuple(lhs_transpose.size() - 1, rhs_transpose.size() - 1); in transposeForBatchMatmul()
229 *lhs = createTransposeOp(*lhs, loc, lhs_transpose, rewriter); in transposeForBatchMatmul()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 121 auto lhs_transpose = lhs_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()
|
| /external/eigen/Eigen/src/Core/products/ |
| D | SelfadjointMatrixMatrix.h | 339 LhsTransposeMapper lhs_transpose(_lhs,lhsStride);
374 pack_lhs_transposed(blockA, lhs_transpose.getSubMapper(i2, k2), actual_kc, actual_mc);
|