Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util.cc361 std::vector<int64> logical_to_physical(layout.minor_to_major_size()); in MakeLogicalToPhysical() local
362 for (int64 physical = 0; physical < logical_to_physical.size(); ++physical) { in MakeLogicalToPhysical()
364 logical_to_physical[logical] = physical; in MakeLogicalToPhysical()
366 return logical_to_physical; in MakeLogicalToPhysical()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc36 std::vector<int64> logical_to_physical = in MakeDescriptors() local
51 for (physical_dim = 0; physical_dim != logical_to_physical[feature_index]; in MakeDescriptors()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc266 std::vector<int64> logical_to_physical = in SourceIndexOfBroadcast() local
274 int64 physical_dim = logical_to_physical[dimension_mapping[i]]; in SourceIndexOfBroadcast()
290 logical_to_physical[dimension_mapping[i]] - min_broadcasted_dimension) { in SourceIndexOfBroadcast()