Searched refs:spatial_dim_start (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 180 int spatial_dim_start, in NeedsReformatTypeAndPermutation() argument
186 spatial_dim_start != default_spatial_dim_start; in NeedsReformatTypeAndPermutation()
193 int spatial_dim_start, int default_batch_dim, in GetReformatTypeAndPermutation() argument
206 permutation_array[default_spatial_dim_start + i] = spatial_dim_start + i; in GetReformatTypeAndPermutation()
208 shape[spatial_dim_start + i]; in GetReformatTypeAndPermutation()
225 const int spatial_dim_start = in FormatToNHWC() local
228 batch_dim, feature_dim, spatial_dim_start, default_batch_dim, in FormatToNHWC()
234 GetReformatTypeAndPermutation(batch_dim, feature_dim, spatial_dim_start, in FormatToNHWC()
|