Searched refs:node_defs (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/profiler/internal/ |
D | tfprof_stats.cc | 194 std::map<string, const NodeDef*> node_defs; in AddGraph() local 203 node_defs[node.name()] = &node; in AddGraph() 205 for (auto it = node_defs.begin(); it != node_defs.end(); it++) { in AddGraph()
|
/external/tensorflow/tensorflow/core/graph/ |
D | graph_constructor.cc | 125 const Options& opts, NodeDefSlice node_defs, const VersionDef* versions, in Construct() argument 135 GraphConstructor c(opts, node_defs, versions, library, g, refiner, in Construct() 144 GraphConstructor(const Options& opts, NodeDefSlice node_defs, in GraphConstructor() argument 152 node_defs_(node_defs), in GraphConstructor() 482 const GraphConstructor::NodeDefSlice& node_defs) { in GetNextIterationNodes() argument 485 for (int n = 0; n < node_defs.size(); ++n) { in GetNextIterationNodes() 486 const NodeDef& node_def = *node_defs[n]; in GetNextIterationNodes() 1220 std::vector<const NodeDef*> node_defs; in ConvertNodeDefsToGraph() local 1222 node_defs.push_back(&n); in ConvertNodeDefsToGraph() 1224 return GraphConstructor::Construct(opts, node_defs, nullptr, nullptr, g, in ConvertNodeDefsToGraph()
|
/external/tensorflow/tensorflow/python/ops/ |
D | while_v2.py | 833 def _resource_input_index(self, tensor_name, input_names, node_defs, argument 870 node_def = node_defs[op_name]
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | virtual_scheduler_test.cc | 1656 const std::vector<const NodeDef*>& node_defs) { in ValidateNodeDefs() argument 1658 std::transform(node_defs.begin(), node_defs.end(), in ValidateNodeDefs()
|