Home
last modified time | relevance | path

Searched refs:dimension_mapping (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc203 absl::Span<const int64> dimension_mapping, in SourceIndexOfTranspose() argument
206 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() argument
257 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()
Dir_array.h143 absl::Span<const int64> dimension_mapping,
154 absl::Span<const int64> dimension_mapping,
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc1072 absl::Span<const int64> dimension_mapping) { in TransposeIsBitcast() argument
1095 ComposePermutations(dimension_mapping, in TransposeIsBitcast()
Dshape_util.h588 absl::Span<const int64> dimension_mapping);