Searched refs:read_var_node (Results 1 – 2 of 2) sorted by relevance
51 read_var_node = []57 read_var_node.append(n)64 read_var_node.append(n)68 self.assertLen(read_var_node, 0)
4170 NodeDef* read_var_node = ctx().optimized_graph->add_node(); in TrySimplify() local4171 read_var_node->set_name(OptimizedNodeName( in TrySimplify()4173 read_var_node->set_op("ReadVariableOp"); in TrySimplify()4174 read_var_node->add_input(gather_node->input(0)); in TrySimplify()4175 read_var_node->set_device(variable_node->device()); in TrySimplify()4179 auto attr = read_var_node->mutable_attr(); in TrySimplify()4197 ctx().node_map->AddNode(read_var_node->name(), read_var_node); in TrySimplify()4198 reduction_node->set_input(0, read_var_node->name()); in TrySimplify()4201 read_var_node->name()); in TrySimplify()