Searched refs:sa_name (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 316 DataType dtype, int sa_id, const string& sa_name, in ConstructScopedAllocatorNode() argument
319 VLOG(2) << "ConstructScopedAllocatorNode " << sa_name; in ConstructScopedAllocatorNode()
320 NodeDefBuilder sa_builder(sa_name, "_ScopedAllocator"); in ConstructScopedAllocatorNode()
322 sa_builder.Attr("sa_name", sa_name); in ConstructScopedAllocatorNode()
330 node_map->AddNode(sa_name, sa_node); in ConstructScopedAllocatorNode()
338 << "^" << sa_name; in ConstructScopedAllocatorNode()
339 nd.from_node_def->add_input(strings::StrCat("^", sa_name)); in ConstructScopedAllocatorNode()
345 node_map->AddOutput(sa_name, nd.from_node_def->name()); in ConstructScopedAllocatorNode()
354 const string& sa_name, const string& sac_name, in BuildSAConcatNode() argument
390 sac_builder.Attr("sa_name", sa_name); in BuildSAConcatNode()
[all …]
|