Searched defs:replace_op (Results 1 – 8 of 8) sorted by relevance
251 Operator replace_op = {std::make_pair(DROPOUT_GEN_MASK, args)}; in GetDropoutGenMaskReplaceOp() local256 static void ReplaceOneOp(const Operator &replace_op, const CNodePtr &node) { in ReplaceOneOp()285 std::vector<Operator> replace_op = GetDropoutGenMaskReplaceOp(cnode); in ReplaceNodeInputOrAttrs() local
82 ForwardOp replace_op() const { return replace_op_; } in replace_op() function
193 ForwardOp replace_op() const { return replace_op_; } in replace_op() function
59 AnfNodePtr replace_op() const { return replace_op_; } in replace_op() function
302 CNodePtr replace_op = func_graph->NewCNode(replace_op_inputs); in ReplaceScalarOp() local
823 auto replace_op = distribute_operator->replace_op(); in ReplaceGatherOps() local
811 static void StepReplaceOp(OperatorVector replace_op, const CNodePtr &node) { in StepReplaceOp()2606 auto replace_op = distribute_operator->replace_op(); in StepReplace() local
773 std::vector<AnfNodePtr> ReplaceOpInput(const Operator &replace_op, const std::string &instance_name, in ReplaceOpInput()