Searched refs:physical_dim (Results 1 – 2 of 2) sorted by relevance
39 auto physical_dim_size = [&](int64 physical_dim) { in MakeDescriptors() argument40 return shape.dimensions(LayoutUtil::Major(shape.layout(), physical_dim)); in MakeDescriptors()50 int64 physical_dim; in MakeDescriptors() local51 for (physical_dim = 0; physical_dim != logical_to_physical[feature_index]; in MakeDescriptors()52 ++physical_dim) { in MakeDescriptors()53 CHECK_LT(physical_dim, shape.dimensions_size()); in MakeDescriptors()54 batch_size *= physical_dim_size(physical_dim); in MakeDescriptors()56 ++physical_dim; // Skip the feature dimension. in MakeDescriptors()57 for (; physical_dim < shape.dimensions_size(); ++physical_dim) { in MakeDescriptors()58 y_size *= physical_dim_size(physical_dim); in MakeDescriptors()
274 int64 physical_dim = logical_to_physical[dimension_mapping[i]]; in SourceIndexOfBroadcast() local276 std::min(min_broadcasted_dimension, physical_dim); in SourceIndexOfBroadcast()278 std::max(max_broadcasted_dimension, physical_dim); in SourceIndexOfBroadcast()