| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime.h | 69 const string graph_handle; member
75 FunctionData(const string& graph_handle, const string& target, in FunctionData()
78 : graph_handle(graph_handle), in FunctionData()
|
| D | worker.cc | 101 s = session->graph_mgr()->Deregister(request->graph_handle()); in DeregisterGraphAsync()
222 request->graph_handle(), step_id, session.get(), request->exec_opts(), in DoRunGraph()
263 const string& graph_handle = request->graph_handle(); in DoPartialRunGraph() local
315 graph_handle, step_id, session.get(), request->exec_opts(), in DoPartialRunGraph()
|
| D | message_wrappers.cc | 309 const string& InMemoryRunGraphRequest::graph_handle() const { in graph_handle() function in tensorflow::InMemoryRunGraphRequest
409 proto_version_->set_graph_handle(graph_handle()); in ToProto()
446 const string& MutableProtoRunGraphRequest::graph_handle() const { in graph_handle() function in tensorflow::MutableProtoRunGraphRequest
447 return request_.graph_handle(); in graph_handle()
569 const string& ProtoRunGraphRequest::graph_handle() const { in graph_handle() function in tensorflow::ProtoRunGraphRequest
570 return request_->graph_handle(); in graph_handle()
|
| D | graph_mgr.cc | 293 DistributedFunctionLibraryRuntime* cluster_flr, string* graph_handle) { in Register() argument
305 *graph_handle = strings::Printf("%016llx", ++next_id_); in Register()
306 item->handle = *graph_handle; in Register()
307 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 | cluster_function_library_runtime.cc | 241 function_data_.push_back(FunctionData(resp->graph_handle(), target, in Instantiate()
276 req->set_graph_handle(function_data->graph_handle); in Run()
|
| D | graph_mgr.h | 82 string* graph_handle);
|
| D | master_session.cc | 266 string graph_handle; member
493 partitions_[i].graph_handle = c->resp.graph_handle(); in DoRegisterPartitions()
682 c->req->set_graph_handle(part.graph_handle); in RunPartitionsHelper()
1100 if (!part.graph_handle.empty()) { in DeregisterPartitions()
1104 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()
133 JNIEnv* env, jclass clazz, jlong graph_handle, jlong op_handle, in Java_org_tensorflow_GraphOperation_dtype() argument
135 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 | 143 // If the registration succeeds, returns an opaque graph_handle to
144 // the master. The master calls RunGraph with graph_handle to
146 string graph_handle = 1; field
153 // The master deregisters the given graph_handle when the graph is no
157 // The worker deregisters a graph_handle automatically according to on
170 // REQUIRED: graph_handle must be returned by a RegisterGraph call
172 string graph_handle = 1; field
202 // The worker executes all subgraphs registered under graph_handle.
222 // search for the graph_handle.
228 // REQUIRED: graph_handle must be returned by a RegisterGraph call
[all …]
|