| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime.h | 76 const string graph_handle; member
85 FunctionData(const string& graph_handle, const string& target, in FunctionData()
89 : graph_handle(graph_handle), in FunctionData()
|
| D | worker.cc | 102 s = session->graph_mgr()->Deregister(request->graph_handle()); in DeregisterGraphAsync()
225 request->graph_handle(), step_id, session.get(), request->exec_opts(), in DoRunGraph()
266 const string& graph_handle = request->graph_handle(); in DoPartialRunGraph() local
318 graph_handle, step_id, session.get(), request->exec_opts(), in DoPartialRunGraph()
|
| D | message_wrappers.cc | 305 const string& InMemoryRunGraphRequest::graph_handle() const { in graph_handle() function in tensorflow::InMemoryRunGraphRequest
405 proto_version_->set_graph_handle(graph_handle()); in ToProto()
442 const string& MutableProtoRunGraphRequest::graph_handle() const { in graph_handle() function in tensorflow::MutableProtoRunGraphRequest
443 return request_.graph_handle(); in graph_handle()
565 const string& ProtoRunGraphRequest::graph_handle() const { in graph_handle() function in tensorflow::ProtoRunGraphRequest
566 return request_->graph_handle(); in graph_handle()
|
| D | graph_mgr.cc | 299 DistributedFunctionLibraryRuntime* cluster_flr, string* graph_handle) { in Register() argument
311 *graph_handle = in Register()
313 item->handle = *graph_handle; in Register()
314 CHECK(table_.insert({*graph_handle, item}).second); in Register()
|
| D | message_wrappers.h | 263 virtual const string& graph_handle() const = 0;
335 const string& graph_handle() const override;
396 const string& graph_handle() const override;
439 const string& graph_handle() const override;
|
| D | graph_mgr.h | 82 string* graph_handle);
|
| D | cluster_function_library_runtime.cc | 242 function_data_.push_back(FunctionData(resp->graph_handle(), target, in Instantiate()
278 req->set_graph_handle(function_data->graph_handle); in Run()
|
| D | master_session.cc | 268 string graph_handle; member
495 partitions_[i].graph_handle = c->resp.graph_handle(); in DoRegisterPartitions()
684 c->req->set_graph_handle(part.graph_handle); in RunPartitionsHelper()
1102 if (!part.graph_handle.empty()) { in DeregisterPartitions()
1106 c->req.set_graph_handle(part.graph_handle); in DeregisterPartitions()
|
| D | message_wrappers_test.cc | 87 EXPECT_EQ("graph_handle", request.graph_handle()); in CheckRunGraphRequest()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | session_jni.cc | 74 JNIEnv* env, jclass clazz, jlong graph_handle) { in Java_org_tensorflow_Session_allocate() argument
75 return Java_org_tensorflow_Session_allocate2(env, clazz, graph_handle, in Java_org_tensorflow_Session_allocate()
80 JNIEnv* env, jclass clazz, jlong graph_handle, jstring target, in Java_org_tensorflow_Session_allocate2() argument
82 if (graph_handle == 0) { in Java_org_tensorflow_Session_allocate2()
86 TF_Graph* graph = reinterpret_cast<TF_Graph*>(graph_handle); in Java_org_tensorflow_Session_allocate2()
|
| D | graph_operation_jni.cc | 82 JNIEnv* env, jclass clazz, jlong graph_handle, jlong op_handle, in Java_org_tensorflow_GraphOperation_shape() argument
84 TF_Graph* graph = requireGraphHandle(env, graph_handle); in Java_org_tensorflow_GraphOperation_shape()
132 JNIEnv* env, jclass clazz, jlong graph_handle, jlong op_handle, in Java_org_tensorflow_GraphOperation_dtype() argument
134 TF_Graph* graph = requireGraphHandle(env, graph_handle); in Java_org_tensorflow_GraphOperation_dtype()
|
| D | graph_operation_builder_jni.cc | 54 JNIEnv* env, jclass clazz, jlong graph_handle, jstring type, jstring name) { in Java_org_tensorflow_GraphOperationBuilder_allocate() argument
55 if (graph_handle == 0) { in Java_org_tensorflow_GraphOperationBuilder_allocate()
60 TF_Graph* graph = reinterpret_cast<TF_Graph*>(graph_handle); in Java_org_tensorflow_GraphOperationBuilder_allocate()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 154 // If the registration succeeds, returns an opaque graph_handle to
155 // the master. The master calls RunGraph with graph_handle to
157 string graph_handle = 1; field
164 // The master deregisters the given graph_handle when the graph is no
168 // The worker deregisters a graph_handle automatically according to on
181 // REQUIRED: graph_handle must be returned by a RegisterGraph call
183 string graph_handle = 1; field
213 // The worker executes all subgraphs registered under graph_handle.
233 // search for the graph_handle.
239 // REQUIRED: graph_handle must be returned by a RegisterGraph call
[all …]
|