Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc496 const NodeDef* const shape_n_node = node; in MaterializeShapes() local
512 auto fanouts = node_map_->GetOutputs(shape_n_node->name()); in MaterializeShapes()
522 if (node_name == shape_n_node->name() && port == port_idx) { in MaterializeShapes()
525 *shape_n_node, strings::StrCat("-matshapes-", port_idx)); in MaterializeShapes()
530 added_node->set_device(shape_n_node->device()); in MaterializeShapes()
538 string ctrl_dep = AddControlDependency(shape_n_node->name(), in MaterializeShapes()
547 if (node_name == shape_n_node->name() && port != port_idx) { in MaterializeShapes()