Searched refs:make_tuple_node (Results 1 – 8 of 8) sorted by relevance
98 auto make_tuple_node = output->cast<CNodePtr>(); in GetForwardOutNodeAndBpropGraph() local99 MS_EXCEPTION_IF_NULL(make_tuple_node); in GetForwardOutNodeAndBpropGraph()101 if (make_tuple_node->size() != input_size) { in GetForwardOutNodeAndBpropGraph()102 …MS_LOG(EXCEPTION) << "The inputs size of make tuple node " << make_tuple_node->DebugString() << " … in GetForwardOutNodeAndBpropGraph()108 const auto &output_node = make_tuple_node->input(forward_output_index); in GetForwardOutNodeAndBpropGraph()116 const auto &bprop_vnode = make_tuple_node->input(bprop_graph_index); in GetForwardOutNodeAndBpropGraph()
474 auto make_tuple_node = utils::cast<CNodePtr>(input_node); in RemoveIfMakeTuple() local475 MS_CHECK_TRUE_MSG(make_tuple_node != nullptr, RET_NULL_PTR, "make_tuple_node is nullptr"); in RemoveIfMakeTuple()476 auto value_node = make_tuple_node->input(0)->cast<ValueNodePtr>(); in RemoveIfMakeTuple()478 …if (value_node->value() != nullptr && (opt::CheckPrimitiveType(make_tuple_node, prim::kPrimMakeTup… in RemoveIfMakeTuple()479 … opt::CheckPrimitiveType(make_tuple_node, opt::kPrimMakeTupleV2))) { in RemoveIfMakeTuple()481 for (size_t j = 1; j < make_tuple_node->inputs().size(); ++j) { in RemoveIfMakeTuple()482 inputs.emplace_back(make_tuple_node->input(j)); in RemoveIfMakeTuple()
174 VectorRef make_tuple_node = VectorRef(outputs); in GetBodyGraphPattern() local177 VectorRef return_node = VectorRef({is_return, make_tuple_node}); in GetBodyGraphPattern()
272 VectorRef make_tuple_node = VectorRef(outputs); in GetBodyGraphPattern() local275 VectorRef return_node = VectorRef({is_return, make_tuple_node}); in GetBodyGraphPattern()
367 VectorRef make_tuple_node = VectorRef(outputs); in GetBodyGraphPattern() local370 VectorRef return_node = VectorRef({is_return, make_tuple_node}); in GetBodyGraphPattern()
1184 auto make_tuple_node = func_graph_->NewCNode(tuple_inputs); in HandleNoEliminateNodes() local1185 make_tuple_node->set_abstract(std::make_shared<abstract::AbstractTuple>(element_abstracts)); in HandleNoEliminateNodes()1186 AttachToOutput(make_tuple_node); in HandleNoEliminateNodes()
885 auto make_tuple_node = switch_layer_cnode->input(kMakeTupleInSwitchLayerIndex); in CreateCallSwitchLayerInputs() local886 MS_EXCEPTION_IF_NULL(make_tuple_node); in CreateCallSwitchLayerInputs()887 auto node = make_tuple_node->cast<CNodePtr>(); in CreateCallSwitchLayerInputs()929 auto abstract = make_tuple_node->abstract(); in CreateCallSwitchLayerInputs()
795 auto make_tuple_node = make_tuple_iter->cast<CNodePtr>(); in GetCaseNodeInput() local811 tuple_out_handle_cache_[make_tuple_node.get()] = tuple_items; in GetCaseNodeInput()