Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dsubgraph.h53 PruneRewrite(const string* endpoint_name, const DeviceAttributes* device_info) in PruneRewrite() argument
54 : endpoint_name_(endpoint_name), device_info_(device_info) {} in PruneRewrite()
63 const string& endpoint_name() { return *endpoint_name_; } in endpoint_name() function
122 ArgFeedRewrite(const string* endpoint_name, in ArgFeedRewrite() argument
124 : PruneRewrite(endpoint_name, device_info), arg_index_(arg_index) {} in ArgFeedRewrite()
143 RetvalFetchRewrite(const string* endpoint_name, in RetvalFetchRewrite() argument
145 : PruneRewrite(endpoint_name, device_info), retval_index_(retval_index) {} in RetvalFetchRewrite()
Dsubgraph.cc64 const string& t = feed_rewrites[i]->endpoint_name(); in FeedInputs()
130 const string& t = fetch_rewrites[i]->endpoint_name(); in FetchOutputs()
245 .Attr("tensor_name", endpoint_name()) in AddNode()
283 .Attr("tensor_name", endpoint_name()) in AddNode()
351 auto result = endpoints.insert(feed_rewrite->endpoint_name()); in RewriteGraphForExecution()
354 feed_rewrite->endpoint_name(), in RewriteGraphForExecution()
360 if (endpoints.count(fetch_rewrite->endpoint_name()) > 0) { in RewriteGraphForExecution()
361 return errors::InvalidArgument(fetch_rewrite->endpoint_name(), in RewriteGraphForExecution()
/external/tensorflow/tensorflow/contrib/slim/python/slim/nets/
Dinception_v1_test.py114 for endpoint_name in endpoints_shapes:
115 expected_shape = endpoints_shapes[endpoint_name]
116 self.assertTrue(endpoint_name in end_points)
117 self.assertListEqual(end_points[endpoint_name].get_shape().as_list(),
Dinception_v2_test.py110 for endpoint_name in endpoints_shapes:
111 expected_shape = endpoints_shapes[endpoint_name]
112 self.assertTrue(endpoint_name in end_points)
113 self.assertListEqual(end_points[endpoint_name].get_shape().as_list(),
Dinception_v3_test.py116 for endpoint_name in endpoints_shapes:
117 expected_shape = endpoints_shapes[endpoint_name]
118 self.assertTrue(endpoint_name in end_points)
119 self.assertListEqual(end_points[endpoint_name].get_shape().as_list(),
/external/tensorflow/tensorflow/python/framework/
Dpython_op_gen_internal.cc601 string endpoint_name; in AddExport() local
603 &endpoint_name); in AddExport()
606 strings::StrAppend(&deprecated_endpoints, "'", endpoint_name, "'"); in AddExport()
610 strings::StrAppend(&names_v1, "'", endpoint_name, "'"); in AddExport()
616 strings::StrAppend(&names, "'", endpoint_name, "'"); in AddExport()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc253 TensorConnectionPruneRewrite(const string* endpoint_name, in TensorConnectionPruneRewrite() argument
255 : subgraph::PruneRewrite(endpoint_name, nullptr /* device_info */), in TensorConnectionPruneRewrite()