| 352 | return out; |
| 353 | } |
| 354 | |
| 355 | core::TensorValue flatten_freq_time_python_order( |
| 356 | core::ModuleBuildContext & ctx, |
| 357 | const core::TensorValue & input) { |
| 358 | auto x = modules::TransposeModule({{0, 1, 3, 2}, input.shape.rank}).build(ctx, input); |
| 359 | x = ensure_contiguous(ctx, x); |
| 360 | return core::reshape_tensor( |
| 361 | ctx, |
| 362 | x, |
| 363 | core::TensorShape::from_dims({input.shape.dims[0], input.shape.dims[1], input.shape.dims[2] * input.shape.dims[3]})); |
| 364 | } |
| 365 | |
| 366 | core::TensorValue unflatten_freq_time_python_order( |
no test coverage detected