Home
last modified time | relevance | path

Searched refs:logical_to_physical (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dlayout_util.cc412 std::vector<int64_t> logical_to_physical(layout.minor_to_major_size()); in MakeLogicalToPhysical() local
413 for (int64_t physical = 0, end = logical_to_physical.size(); physical < end; in MakeLogicalToPhysical()
416 logical_to_physical[logical] = physical; in MakeLogicalToPhysical()
418 return logical_to_physical; in MakeLogicalToPhysical()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Dir_array.cc382 std::vector<int64_t> logical_to_physical = in SourceIndexOfBroadcast() local
390 int64_t physical_dim = logical_to_physical[dimension_mapping[i]]; in SourceIndexOfBroadcast()
406 logical_to_physical[dimension_mapping[i]] - min_broadcasted_dimension) { in SourceIndexOfBroadcast()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc600 auto logical_to_physical = in HandleSort() local
602 TF_RET_CHECK(sort->sort_dimension() < logical_to_physical.size()); in HandleSort()
604 logical_to_physical[sort->sort_dimension()]; in HandleSort()