Searched defs:default_batch_dim (Results 1 – 1 of 1) sorted by relevance
181 int default_batch_dim, in NeedsReformatTypeAndPermutation()193 int spatial_dim_start, int default_batch_dim, in GetReformatTypeAndPermutation()220 int default_batch_dim, int default_feature_dim, in FormatToNHWC()