Home
last modified time | relevance | path

Searched refs:sac_name (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc354 const string& sa_name, const string& sac_name, in BuildSAConcatNode() argument
357 VLOG(2) << "BuildSAConcatNode " << sac_name; in BuildSAConcatNode()
386 NodeDefBuilder sac_builder(sac_name, "_ScopedAllocatorConcat"); in BuildSAConcatNode()
387 VLOG(2) << "New sac_name " << sac_name << " shape " in BuildSAConcatNode()
399 node_map->AddNode(sac_name, sac_node); in BuildSAConcatNode()
400 node_map->AddOutput(sa_name, sac_name); in BuildSAConcatNode()
412 const string& op_name, const string& sac_name, in BuildReplacementOp() argument
428 op_builder.Input(sac_name, 0, dtype); in BuildReplacementOp()
432 node_map->AddOutput(sac_name, sa_op_name); in BuildReplacementOp()
626 string sac_name = strings::StrCat("scoped_allocator_concat_", sa_id, "_", in Rewrite() local
[all …]