Searched refs:final_split_spatial_dim_positioning (Results 1 – 1 of 1) sorted by relevance
147 absl::Span<const int64_t> final_split_spatial_dim_positioning,2814 absl::Span<const int64_t> final_split_spatial_dim_positioning, in TransposeAndMergeBatch() argument2816 const int64_t spatial_dim_count = final_split_spatial_dim_positioning.size(); in TransposeAndMergeBatch()2818 if (final_split_spatial_dim_positioning.size() > 1) { in TransposeAndMergeBatch()