Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_stats.cc194 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/
Dgraph_constructor.cc125 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/
Dwhile_v2.py833 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/
Dvirtual_scheduler_test.cc1656 const std::vector<const NodeDef*>& node_defs) { in ValidateNodeDefs() argument
1658 std::transform(node_defs.begin(), node_defs.end(), in ValidateNodeDefs()