| 825 | } |
| 826 | |
| 827 | core::TensorValue graph_tsconformer( |
| 828 | core::ModuleBuildContext & ctx, |
| 829 | std::vector<GraphConstant> & constants, |
| 830 | core::TensorValue x, |
| 831 | const std::unordered_map<std::string, Param> & params) { |
| 832 | x = graph_run_dual_encoder(ctx, constants, x, params, "TSConformer.encoders.0"); |
| 833 | x = graph_run_downsampled_encoder(ctx, constants, x, params, "TSConformer.encoders.1", 2); |
| 834 | x = graph_run_downsampled_encoder(ctx, constants, x, params, "TSConformer.encoders.2", 2); |
| 835 | x = graph_run_dual_encoder(ctx, constants, x, params, "TSConformer.encoders.3"); |
| 836 | return x; |
| 837 | } |
| 838 | |
| 839 | core::TensorValue graph_subpixel_conv( |
| 840 | core::ModuleBuildContext & ctx, |
no test coverage detected