Searched defs:physical_dim (Results 1 – 2 of 2) sorted by relevance
75 auto physical_dim_size = [&](int64 physical_dim) { in MakeBatchNormDescriptors()86 int64 physical_dim; in MakeBatchNormDescriptors() local
361 int64 physical_dim = logical_to_physical[dimension_mapping[i]]; in SourceIndexOfBroadcast() local