Searched refs:weight_index (Results 1 – 3 of 3) sorted by relevance
40 …ePtr GetRealLstmWeightNode(const FuncGraphPtr &graph, const CNodePtr &cnode, size_t weight_index) { in GetRealLstmWeightNode() argument46 auto weight_node = cnode->input(weight_index); in GetRealLstmWeightNode()64 (void)manager->Replace(cnode->input(weight_index), weight_node); in GetRealLstmWeightNode()
159 int64_t weight_index = GetValue<int64_t>(weight_index_value_node->value()); in DoFusion() local160 indices.push_back(weight_index); in DoFusion()
609 const size_t weight_index = 2; in FetchInfoForQuantExport() local615 auto weight = root_node->input(weight_index); in FetchInfoForQuantExport()630 auto weight_node = cnode->input(weight_index); in FetchInfoForQuantExport()