Searched defs:replace_op (Results 1 – 5 of 5) sorted by relevance
264 Operator replace_op = {std::make_pair(DROPOUT_GEN_MASK, args)}; in GetDropoutGenMaskReplaceOp() local269 static void ReplaceOneOp(const Operator &replace_op, const CNodePtr &node) { in ReplaceOneOp()298 std::vector<Operator> replace_op = GetDropoutGenMaskReplaceOp(); in ReplaceNodeInputOrAttrs() local
115 ForwardOp replace_op() const { return replace_op_; } in replace_op() function
60 AnfNodePtr replace_op() const { return replace_op_; } in replace_op() function
186 std::vector<AnfNodePtr> ReplaceOpInput(const Operator &replace_op, const std::string &instance_name, in ReplaceOpInput()
715 void StepReplaceOp(OperatorVector replace_op, const CNodePtr &node) { in StepReplaceOp()2430 OperatorVector replace_op = distribute_operator->replace_op(); in StepReplace() local