Searched refs:spatial_dim_start (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | legalize_hlo.cc | 501 int spatial_dim_start, in NeedsReformatTypeAndPermutation() argument
507 spatial_dim_start != default_spatial_dim_start; in NeedsReformatTypeAndPermutation()
514 int spatial_dim_start, int default_batch_dim, in GetReformatTypeAndPermutation() argument
527 permutation_array[default_spatial_dim_start + i] = spatial_dim_start + i; in GetReformatTypeAndPermutation()
529 shape[spatial_dim_start + i]; in GetReformatTypeAndPermutation()
546 const int spatial_dim_start = spatial_dimensions.front(); in FormatToNHWC() local
548 batch_dim, feature_dim, spatial_dim_start, default_batch_dim, in FormatToNHWC()
554 GetReformatTypeAndPermutation(batch_dim, feature_dim, spatial_dim_start, in FormatToNHWC()
|