Searched refs:assign_node (Results 1 – 4 of 4) sorted by relevance
1291 NodeDef* assign_node = optimized_graph->mutable_node(idx); in RelaxAssignNodes() local1292 (*assign_node->mutable_attr())["_grappler_relax_allocator_constraints"] in RelaxAssignNodes()1325 const NodeDef& assign_node = graph.node(i); in FindAssignNodesToRelax() local1327 if (optimized_nodes.find(&assign_node) == optimized_nodes.end()) { in FindAssignNodesToRelax()1329 optimized_nodes.insert(&assign_node); in FindAssignNodesToRelax()1330 assign_nodes_in_fanout.push_back(&assign_node); in FindAssignNodesToRelax()1350 CrossesTaskOrCpuGpuBoundary(*fanout_node, assign_node))) { in FindAssignNodesToRelax()
86 NodeDef* assign_node; in TestSinglePartition() local103 assign_node = CreateNode("w/part_1/Assign", "Assign", in TestSinglePartition()142 CreateNode(shared_init_name, "NoOp", {assign_node}, &graph_def, true); in TestSinglePartition()
290 assign_node = node.body[1]294 anno.getanno(assign_node, anno.Basic.ORIGIN).loc.lineno, 103)
941 AstNode* assign_node = AddNode( in ParseRule() local948 id->sibling = assign_node; in ParseRule()962 assign_node->sibling = type; in ParseRule()