Home
last modified time | relevance | path

Searched defs:func_node (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_compiler/inliner/
Dfunc_inliner.cc117 ir::FunctionNodePtr func_node = byteCodeParser->Parse(); in Visit_() local
150 void FuncInlineDetector::EvolvingFunction(const ir::FunctionNodePtr &func_node, const ir::NodePtrLi… in EvolvingFunction() argument
226 auto func_node = func->cast<ir::FunctionNodePtr>(); in Mutate_() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/auto_grad/
Dfunction_node.cc50 auto func_node = grad_fn.cast<grad::FunctionNodePtr>(); in PostBpropFunctionToEdges() local
134 auto func_node = FunctionNode::CreateFunctionNode(tensor, prim, out, inputs); in GetOrCreateFunctionNode() local
144 auto func_node = std::make_shared<FunctionNode>(tensor, prim, out); in CreateFunctionNode() local
163 auto func_node = CreateFunctionNode(out, prim, out, inputs); in RecordPrimitive() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_compiler/
Dcompiler.cc213 ir::FunctionNodePtr func_node = byteCodeParser->Parse(); in Compile() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/
Dvariable.h188 BackwardNodePtr func_node() const { return func_node_; } in func_node() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/function/
Dfunc_grad.cc208 auto func_node = emitter->NewFuncNode(op_inputs_[i], input_abstract_[i], grad_type_[i]); in PreProcess() local
304 auto func_node = std::make_shared<BackwardNode>("input" + std::to_string(i)); in FuncGrad() local
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/
Dutils.cc363 auto func_node = call_node->input(0); in GetFuncGraphsFromCallNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/
Dir_bprop.cc284 auto func_node = std::make_shared<IrFunctionNode>(ad_param_->tape_, zeros_like_dout); in AddParameterNode() local
640 auto func_node = std::make_shared<IrFunctionNode>(ad_param_->tape_, zeros_like_dout); in CreateParameterAdjoint() local
Dir_grad.cc206 auto func_node = std::make_shared<IrFunctionNode>(ad_param()->tape_, zeros_like_dout); in IrGrad() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_capture/
Dgraph.cc373 static bool SkipGuardInlinedFunc(ValueNode *func_node) { in SkipGuardInlinedFunc()
Dgraph_build.cc1639 py::object GraphBuilder::GetFuncInfo(ValueNode *func_node) { in GetFuncInfo()
2072 ValueNode *func_node = this->NewValueNode(AObject::Convert(new_func), LOAD_CONST, -1, {}); in ReplaceCall() local
2669 auto func_node = call_node->input(0); in SetMixedPrecisionType() local
3035 auto func_node = node->input(0); in IsShapeOrDtypeRelatedNode() local
Dspecial_func_infer.cc786 auto func_node = call_node->input(0); in InferMappingGet() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/static_analysis/
Dstatic_analysis.cc240 const AnfNodePtr &func_node = nullptr) { in ConvertToPyInterpretCall()
804 const auto &func_node = cnode->input(0); in InterpretedNodeCall() local
836 auto &func_node = cnode->input(0); in GetCNodeOperatorAbstract() local
Devaluator.cc400 const AnfNodePtr &func_node = fg->get_return(); in LaunchRecursiveEval() local
Dauto_monad.cc526 const AnfNodePtr &func_node = cnode->input(0); in TracePartialCallEffectInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dcontrol_node_parser.cc1605 auto &func_node = inputs[kPartialFuncGraphPos]; in ParseDeviceContextForPartialNode() local
2036 auto &func_node = inputs[kPartialFuncGraphPos]; in ParseFormalToRealParameter() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/
Dcommon.cc1424 mindspore::pijit::ir::FunctionNodePtr func_node = byteCodeParser->Parse(); in test_graph_ir_code_gen() local
/third_party/mindspore/mindspore-src/source/mindspore/core/load_mindir/
Dload_model.cc2399 auto func_node = GetAnfNode(attr_proto.s()); in BuildAbstractFunction() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/parse/
Dparse.cc4875 py::list ParseFunctionAst::GetArgs(const py::object &func_node) { in GetArgs()
4880 py::list ParseFunctionAst::GetArgsDefaultValues(const py::object &func_node) { in GetArgsDefaultValues()