Searched refs:load_cnode (Results 1 – 3 of 3) sorted by relevance
45 auto load_cnode = load_node->cast<CNodePtr>(); in operator ()() local48 auto &input_load = load_cnode->input(kFirstInputIndex); in operator ()()53 auto monad = load_cnode->input(kSecondInputIndex); in operator ()()55 auto fg = load_cnode->func_graph(); in operator ()()58 new_load->set_abstract(load_cnode->abstract()); in operator ()()59 new_load->set_scope(load_cnode->scope()); in operator ()()
1333 auto load_cnode = func_graph_->NewCNode({load_prim, ref, u}); in MakeLoad() local1336 load_cnode->set_user_data(primitive_target, std::make_shared<std::string>(target)); in MakeLoad()1340 load_cnode->set_abstract(ref_abs->CloneAsTensor()); in MakeLoad()1341 return load_cnode; in MakeLoad()
726 auto load_cnode = anf->cast<CNodePtr>(); in GetNewCNodeInputs() local727 para->set_name(load_cnode->input(kFirstDataInputIndex)->fullname_with_scope()); in GetNewCNodeInputs()