Searched refs:sa_node (Results 1 – 2 of 2) sorted by relevance
580 const NodeDef* sa_node = nullptr; in TEST_F() local583 sa_node = &node; in TEST_F()587 ASSERT_NE(sa_node, nullptr); in TEST_F()590 for (NodeDef* sa_output : node_map.GetOutputs(sa_node->name())) { in TEST_F()
525 NodeDef* sa_node = graph->add_node(); in ConstructScopedAllocatorNode() local526 LOG_WARNING_AND_RETURN_IF_ERROR(sa_builder.Finalize(sa_node)); in ConstructScopedAllocatorNode()527 node_map->AddNode(sa_name, sa_node); in ConstructScopedAllocatorNode()573 sa_node->add_input( in ConstructScopedAllocatorNode()579 << sa_node->name(); in ConstructScopedAllocatorNode()