Searched refs:logical_to_physical (Results 1 – 3 of 3) sorted by relevance
350 std::vector<int64> logical_to_physical(layout.minor_to_major_size()); in MakeLogicalToPhysical() local351 for (int64 physical = 0, end = logical_to_physical.size(); physical < end; in MakeLogicalToPhysical()354 logical_to_physical[logical] = physical; in MakeLogicalToPhysical()356 return logical_to_physical; in MakeLogicalToPhysical()
72 std::vector<int64> logical_to_physical = in MakeBatchNormDescriptors() local87 for (physical_dim = 0; physical_dim != logical_to_physical[feature_index]; in MakeBatchNormDescriptors()
353 std::vector<int64> logical_to_physical = in SourceIndexOfBroadcast() local361 int64 physical_dim = logical_to_physical[dimension_mapping[i]]; in SourceIndexOfBroadcast()377 logical_to_physical[dimension_mapping[i]] - min_broadcasted_dimension) { in SourceIndexOfBroadcast()