Searched refs:assign_node (Results 1 – 2 of 2) sorted by relevance
1266 NodeDef* assign_node = optimized_graph->mutable_node(assign_idx); in RelaxAllocatorConstraints() local1267 (*assign_node->mutable_attr())["_grappler_relax_allocator_constraints"] in RelaxAllocatorConstraints()1279 const NodeDef& assign_node = optimized_graph->node(i); in RelaxAllocatorConstraints() local1281 if (optimized_nodes.find(&assign_node) == optimized_nodes.end()) { in RelaxAllocatorConstraints()1283 optimized_nodes.insert(&assign_node); in RelaxAllocatorConstraints()1284 assign_nodes_in_fanout.push_back(&assign_node); in RelaxAllocatorConstraints()1304 CrossesTaskOrCpuGpuBoundary(*fanout_node, assign_node))) { in RelaxAllocatorConstraints()
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()