Searched refs:physical_dim (Results 1 – 2 of 2) sorted by relevance
75 auto physical_dim_size = [&](int64 physical_dim) { in MakeBatchNormDescriptors() argument76 return shape.dimensions(LayoutUtil::Major(shape.layout(), physical_dim)); in MakeBatchNormDescriptors()86 int64 physical_dim; in MakeBatchNormDescriptors() local87 for (physical_dim = 0; physical_dim != logical_to_physical[feature_index]; in MakeBatchNormDescriptors()88 ++physical_dim) { in MakeBatchNormDescriptors()89 CHECK_LT(physical_dim, shape.dimensions_size()); in MakeBatchNormDescriptors()90 batch_size *= physical_dim_size(physical_dim); in MakeBatchNormDescriptors()92 ++physical_dim; // Skip the feature dimension. in MakeBatchNormDescriptors()93 for (; physical_dim < shape.dimensions_size(); ++physical_dim) { in MakeBatchNormDescriptors()94 y_size *= physical_dim_size(physical_dim); in MakeBatchNormDescriptors()
361 int64 physical_dim = logical_to_physical[dimension_mapping[i]]; in SourceIndexOfBroadcast() local363 std::min(min_broadcasted_dimension, physical_dim); in SourceIndexOfBroadcast()365 std::max(max_broadcasted_dimension, physical_dim); in SourceIndexOfBroadcast()