| 1563 | } |
| 1564 | |
| 1565 | void Reduce::init_kern_sched_shape(const TensorShape& ishp, const TensorShape& oshp) { |
| 1566 | OutTensorShapeExtender extender(ishp, oshp); |
| 1567 | auto&& canonized_oshp = extender.get(); |
| 1568 | m_kern_scheduler->init_shapes( |
| 1569 | static_cast<megdnn::Reduce*>(megdnn_opr()), comp_node(), input(0)->dtype(), |
| 1570 | m_param.mode, ishp, canonized_oshp, m_param.data_type); |
| 1571 | } |
| 1572 | |
| 1573 | cg::OperatorNodeBase::OprEventCallback Reduce::get_opr_event_callback() { |
| 1574 | auto on_mem_status_changed = [this]() { |
nothing calls this directly
no test coverage detected