Searched refs:sa_name (Results 1 – 1 of 1) sorted by relevance
513 DataType dtype, int sa_id, const string& sa_name, in ConstructScopedAllocatorNode() argument516 VLOG(2) << "ConstructScopedAllocatorNode " << sa_name; in ConstructScopedAllocatorNode()517 NodeDefBuilder sa_builder(sa_name, "_ScopedAllocator"); in ConstructScopedAllocatorNode()519 sa_builder.Attr("sa_name", sa_name); in ConstructScopedAllocatorNode()527 node_map->AddNode(sa_name, sa_node); in ConstructScopedAllocatorNode()549 << "^" << sa_name; in ConstructScopedAllocatorNode()550 nd.from_node_def->add_input(strings::StrCat("^", sa_name)); in ConstructScopedAllocatorNode()556 node_map->AddOutput(sa_name, nd.from_node_def->name()); in ConstructScopedAllocatorNode()570 << " in the fanout of " << sa_name; in ConstructScopedAllocatorNode()575 node_map->AddOutput(inputs_to_first[i].from_node_def->name(), sa_name); in ConstructScopedAllocatorNode()[all …]