Searched refs:transpose_shape (Results 1 – 3 of 3) sorted by relevance
98 auto transpose_shape = in Transpose() local103 auto result_type = RankedTensorType::get(transpose_shape, elem_type); in Transpose()
115 Shape transpose_shape, in MakeTransposeHlo()118 HloInstruction::CreateTranspose(transpose_shape, operand, dimensions)); in MakeTransposeHlo()
1109 std::vector<int64_t> transpose_shape(permutation.size()); in matchAndRewrite() local1111 transpose_shape[it.index()] = reshaped_shape[it.value()]; in matchAndRewrite()1115 op.getLoc(), RankedTensorType::get(transpose_shape, element_ty), in matchAndRewrite()