Searched refs:transpose_dimensions (Results 1 – 3 of 3) sorted by relevance
159 std::vector<int64> transpose_dimensions(shape.dimensions_size());160 std::iota(transpose_dimensions.begin(), transpose_dimensions.end(), 1);161 std::reverse(transpose_dimensions.begin(), transpose_dimensions.end());162 transpose_dimensions.insert(transpose_dimensions.begin() + 1, 1, 0);165 XlaOp transposed_result = Transpose(result, transpose_dimensions);
135 const auto& transpose_dimensions = transpose.dimensions(); in FoldTransposeIntoConvolution() local142 transpose_dimensions[dnums.input_batch_dimension()]); in FoldTransposeIntoConvolution()144 transpose_dimensions[dnums.input_feature_dimension()]); in FoldTransposeIntoConvolution()147 input_spatial_dimension = transpose_dimensions[input_spatial_dimension]; in FoldTransposeIntoConvolution()158 const auto& transpose_dimensions = transpose.dimensions(); in FoldTransposeIntoConvolution() local165 transpose_dimensions[dnums.kernel_input_feature_dimension()]); in FoldTransposeIntoConvolution()167 transpose_dimensions[dnums.kernel_output_feature_dimension()]); in FoldTransposeIntoConvolution()170 kernel_spatial_dimension = transpose_dimensions[kernel_spatial_dimension]; in FoldTransposeIntoConvolution()
299 std::vector<int64> transpose_dimensions(batch_dimensions.begin(), in NormalizeDotOperandToBatchMajorAndContractingMinor() local304 transpose_dimensions.push_back(i); in NormalizeDotOperandToBatchMajorAndContractingMinor()307 transpose_dimensions.insert(transpose_dimensions.end(), in NormalizeDotOperandToBatchMajorAndContractingMinor()310 return MakeTransposeHlo(dot_operand, transpose_dimensions); in NormalizeDotOperandToBatchMajorAndContractingMinor()2974 auto transpose_dimensions = arg->dimensions(); in HandleReduce() local2977 new_reduce_dimensions.push_back(transpose_dimensions[dim]); in HandleReduce()