| 209 | } |
| 210 | |
| 211 | cg::OperatorNodeBase::NodeProp* LoopGrad::AssignorGradOpr::do_make_node_prop() const { |
| 212 | auto prop = Super::do_make_node_prop(); |
| 213 | using DT = NodeProp::DepType; |
| 214 | if (input().size() == 1) { |
| 215 | prop->reset_dep_type(input(), {DT::SHAPE}); |
| 216 | } else { |
| 217 | prop->reset_dep_type(input(), {DT::DEV_VALUE, DT::SHAPE}); |
| 218 | } |
| 219 | return prop; |
| 220 | } |
| 221 | |
| 222 | void LoopGrad::AssignorGradOpr::init_assignee_info( |
| 223 | const VarNodeArray& assignees, SymbolVar loss) { |
nothing calls this directly
no test coverage detected