Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer.cc881 const string& depended_node, in AddNodePermNHWCToNCHW() argument
888 *const_node->add_input() = AsControlDependency(depended_node); in AddNodePermNHWCToNCHW()
893 const string& depended_node, in AddNodePermNCHWToNHWC() argument
900 *const_node->add_input() = AsControlDependency(depended_node); in AddNodePermNCHWToNHWC()
909 string depended_node; in GetOrAddNodePermNHWCToNCHW() local
911 depended_node = input; in GetOrAddNodePermNHWCToNCHW()
914 depended_node = NodeName(input_node->input(0)); in GetOrAddNodePermNHWCToNCHW()
917 AddNodePermNHWCToNCHW(base_name, depended_node, node_->device()); in GetOrAddNodePermNHWCToNCHW()
1340 const string& depended_node) { in AddNodeShapeConst() argument
1361 *node->add_input() = AsControlDependency(depended_node); in AddNodeShapeConst()