Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer.cc648 string const_name = GetOrAddNodePermNHWCToNCHW(pos); in AddLayoutTransposeToInputs() local
652 node_name, node_->input(pos), const_name, dtype, in AddLayoutTransposeToInputs()
665 string const_name = GetOrAddNodePermNCHWToNHWC(); in AddTransformToOutputs() local
690 added_node_name, input, const_name, dtype, in AddTransformToOutputs()
845 const string& const_name, DataType data_type, in AddNodeTranspose() argument
852 *node->add_input() = const_name; in AddNodeTranspose()
905 string const_name; in GetOrAddNodePermNHWCToNCHW() local
918 const_name = const_node->name(); in GetOrAddNodePermNHWCToNCHW()
920 const_name = LayoutOptimizerNode(kPermNHWCToNCHW); in GetOrAddNodePermNHWCToNCHW()
922 return const_name; in GetOrAddNodePermNHWCToNCHW()
[all …]
Dconstant_folding.cc471 const string const_name = OptimizedNodeName( in MaterializeShapes() local
473 if (node_map_->GetNode(const_name) == nullptr) { in MaterializeShapes()
475 added_node->set_name(const_name); in MaterializeShapes()
490 *output->mutable_input(k) = const_name; in MaterializeShapes()
491 node_map_->AddOutput(const_name, output->name()); in MaterializeShapes()
639 string const_name = in MaterializeBroadcastGradientArgs() local
641 out[j] = node_map_->GetNode(const_name); in MaterializeBroadcastGradientArgs()
645 CreateNodeDef(const_name, TensorValue(&value), out[j])); in MaterializeBroadcastGradientArgs()
647 node_map_->AddNode(const_name, out[j]); in MaterializeBroadcastGradientArgs()
651 node_map_->AddOutput(NodeName(ctrl_dep), const_name); in MaterializeBroadcastGradientArgs()
[all …]