Searched refs:endpoint_name (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | subgraph.h | 53 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()
|
| D | subgraph.cc | 63 const string& t = feed_rewrites[i]->endpoint_name(); in FeedInputs()
129 const string& t = fetch_rewrites[i]->endpoint_name(); in FetchOutputs()
244 .Attr("tensor_name", endpoint_name()) in AddNode()
282 .Attr("tensor_name", endpoint_name()) in AddNode()
350 auto result = endpoints.insert(feed_rewrite->endpoint_name()); in RewriteGraphForExecution()
353 feed_rewrite->endpoint_name(), in RewriteGraphForExecution()
359 if (endpoints.count(fetch_rewrite->endpoint_name()) > 0) { in RewriteGraphForExecution()
360 return errors::InvalidArgument(fetch_rewrite->endpoint_name(), in RewriteGraphForExecution()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen_internal.cc | 622 string endpoint_name; in AddExport() local
624 &endpoint_name); in AddExport()
627 strings::StrAppend(&deprecated_endpoints, "'", endpoint_name, "'"); in AddExport()
631 strings::StrAppend(&names_v1, "'", endpoint_name, "'"); in AddExport()
637 strings::StrAppend(&names, "'", endpoint_name, "'"); in AddExport()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.cc | 312 TensorConnectionPruneRewrite(const string* endpoint_name, in TensorConnectionPruneRewrite() argument
314 : subgraph::PruneRewrite(endpoint_name, nullptr /* device_info */), in TensorConnectionPruneRewrite()
|