Searched refs:default_spatial_dim_start (Results 1 – 1 of 1) sorted by relevance
504 int default_spatial_dim_start) const { in NeedsReformatTypeAndPermutation()507 spatial_dim_start != default_spatial_dim_start; in NeedsReformatTypeAndPermutation()516 int default_spatial_dim_start, in GetReformatTypeAndPermutation() argument527 permutation_array[default_spatial_dim_start + i] = spatial_dim_start + i; in GetReformatTypeAndPermutation()528 transposed_shape[default_spatial_dim_start + i] = in GetReformatTypeAndPermutation()542 int default_spatial_dim_start, int num_spatial_dims, in FormatToNHWC() argument549 default_feature_dim, default_spatial_dim_start)) { in FormatToNHWC()556 default_spatial_dim_start, in FormatToNHWC()