Searched refs:dimension_mapping (Results 1 – 4 of 4) sorted by relevance
208 absl::Span<const int64> dimension_mapping, in SourceIndexOfTranspose() argument211 Permute(dimension_mapping, multidim()); in SourceIndexOfTranspose()215 ShapeUtil::TransposeIsBitcast(operand_shape, shape, dimension_mapping)) { in SourceIndexOfTranspose()253 absl::Span<const int64> dimension_mapping, in SourceIndexOfBroadcast() argument258 source_index[i] = multidim_[dimension_mapping[i]]; in SourceIndexOfBroadcast()275 int64 physical_dim = logical_to_physical[dimension_mapping[i]]; in SourceIndexOfBroadcast()291 logical_to_physical[dimension_mapping[i]] - min_broadcasted_dimension) { in SourceIndexOfBroadcast()
139 absl::Span<const int64> dimension_mapping,150 absl::Span<const int64> dimension_mapping,
1099 absl::Span<const int64> dimension_mapping) { in TransposeIsBitcast() argument1122 ComposePermutations(dimension_mapping, in TransposeIsBitcast()
615 absl::Span<const int64> dimension_mapping);