Searched refs:sas_name (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 442 const string& sas_name, const string& sa_name, in BuildSplitNode() argument
444 VLOG(2) << "new ScopedAllocatorSplit " << sas_name; in BuildSplitNode()
445 NodeDefBuilder sas_builder(sas_name, "_ScopedAllocatorSplit"); in BuildSplitNode()
457 node_map->AddNode(sas_name, sas_node); in BuildSplitNode()
458 node_map->AddOutput(sa_op_name, sas_name); in BuildSplitNode()
472 const string& op_name, const string& sas_name) { in RewireSubgraph() argument
511 *n->mutable_input(i) = strings::StrCat("^", sas_name); in RewireSubgraph()
515 *n->mutable_input(i) = strings::StrCat(sas_name, ":", op_idx); in RewireSubgraph()
518 node_map->AddOutput(sas_name, n->name()); in RewireSubgraph()
640 string sas_name = strings::StrCat("scoped_allocator_split_", sa_id, "_", in Rewrite() local
[all …]
|