| 563 | int64_t repeat) { |
| 564 | auto bct = modules::TransposeModule({{0, 2, 1}, 3}).build(ctx, input); |
| 565 | bct = adjacent_repeat_frames_bct(ctx, bct, repeat); |
| 566 | return modules::TransposeModule({{0, 2, 1}, 3}).build(ctx, bct); |
| 567 | } |
| 568 | |
| 569 | core::TensorValue reshape_codebook_indices( |
| 570 | core::ModuleBuildContext & ctx, |
| 571 | const core::TensorValue & codes_bqt, |
| 572 | int64_t quantizer) { |
| 573 | auto selected = modules::SliceModule({1, quantizer, 1}).build(ctx, codes_bqt); |
| 574 | selected = core::ensure_backend_addressable_layout(ctx, modules::TransposeModule({{0, 2, 1}, 3}).build(ctx, selected)); |
| 575 | return core::reshape_tensor( |
| 576 | ctx, |
| 577 | selected, |
no test coverage detected