| 587 | } |
| 588 | |
| 589 | void VirtualGrad::init_output_static_infer_desc() { |
| 590 | using namespace cg::static_infer; |
| 591 | auto&& mgr = owner_graph()->static_infer_manager(); |
| 592 | auto ovar = output(0), ivar = input(1); |
| 593 | mgr.register_shape_infer(ovar, ShapeInferDesc::make_identity(ivar)); |
| 594 | } |
| 595 | |
| 596 | void VirtualGrad::on_output_comp_node_stream_changed() {} |
| 597 |
nothing calls this directly
no test coverage detected