Searched defs:partial_node (Results 1 – 6 of 6) sorted by relevance
309 int PartialGraphIndex(const CNodeT *partial_node) { in PartialGraphIndex()314 int InferShapePass::CopyPartialShapeToSubGraph(const CNodeT *partial_node, MetaGraphT *graph) { in CopyPartialShapeToSubGraph()345 int InferShapePass::RestoreSubGraphInput(const CNodeT *partial_node, MetaGraphT *graph) { in RestoreSubGraphInput()357 int InferShapePass::InferPartialNode(const CNodeT *partial_node, MetaGraphT *graph) { in InferPartialNode()
37 auto partial_node = dyn_cast<CNode>(inputs.at(kSwitchBranchIndex)); in AddOutputAndCallerToMap() local
627 int Scheduler::RestoreSubGraphInput(const lite::LiteGraph::Node *partial_node) { in RestoreSubGraphInput()644 int Scheduler::CopyPartialShapeToSubGraph(const lite::LiteGraph::Node *partial_node) { in CopyPartialShapeToSubGraph()1379 bool Scheduler::IsControlFlowPattern(const lite::LiteGraph::Node &partial_node) { in IsControlFlowPattern()
236 …auto partial_node = kernel::LiteKernelUtil::GetInputsSpecificNode(node, schema::PrimitiveType_Part… in CompileArrowThroughPartialCall() local
468 bool ExistGraphCaller(const AnfNodePtr &partial_node) { in ExistGraphCaller()772 auto partial_node = graph->NewCNode(partial_inputs); in CreateSwitchInput() local796 auto partial_node = node->cast<CNodePtr>(); in CreateCallSwitchInputs() local904 auto partial_node = partial_idx->cast<CNodePtr>(); in CreateCallSwitchLayerInputs() local955 auto partial_node = attr_input->cast<CNodePtr>(); in CreateSwitchOrPartialNode() local1968 std::string SessionBasic::AddPartialParametersMap(const AnfNodePtr &partial_node) { in AddPartialParametersMap()
70 const auto &partial_node = switch_inputs[i]; in FetchParameterBySwitchNode() local882 const auto &partial_node = input->cast<CNodePtr>(); in FetchValueNodeBySwitchNode() local