Searched refs:dimension_mapping (Results 1 – 4 of 4) sorted by relevance
203 absl::Span<const int64> dimension_mapping, in SourceIndexOfTranspose() argument206 Permute(dimension_mapping, multidim()); in SourceIndexOfTranspose()210 ShapeUtil::TransposeIsBitcast(operand_shape, shape, dimension_mapping)) { in SourceIndexOfTranspose()252 absl::Span<const int64> dimension_mapping, in SourceIndexOfBroadcast() argument257 source_index[i] = multidim_[dimension_mapping[i]]; in SourceIndexOfBroadcast()274 int64 physical_dim = logical_to_physical[dimension_mapping[i]]; in SourceIndexOfBroadcast()290 logical_to_physical[dimension_mapping[i]] - min_broadcasted_dimension) { in SourceIndexOfBroadcast()
143 absl::Span<const int64> dimension_mapping,154 absl::Span<const int64> dimension_mapping,
1072 absl::Span<const int64> dimension_mapping) { in TransposeIsBitcast() argument1095 ComposePermutations(dimension_mapping, in TransposeIsBitcast()
588 absl::Span<const int64> dimension_mapping);