| 46 | } |
| 47 | |
| 48 | void init_output_static_infer_desc() override { |
| 49 | using namespace cg::static_infer; |
| 50 | auto&& mgr = owner_graph()->static_infer_manager(); |
| 51 | auto infer_shape = [this](TensorShape& dst, const InpVal& inp) { |
| 52 | size_t n = inp.val.at(0).shape().total_nr_elems(); |
| 53 | dst = TensorShape{n * m_scale}; |
| 54 | return true; |
| 55 | }; |
| 56 | mgr.register_shape_infer( |
| 57 | output(0), |
| 58 | {SourceType::DEP, {{input(0), DepType::SHAPE}}, infer_shape}); |
| 59 | } |
| 60 | |
| 61 | public: |
| 62 | SublinearBadOpr( |
nothing calls this directly
no test coverage detected