| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime.h | 59 const string graph_handle; member
65 FunctionData(const string& graph_handle, const string& target, in FunctionData()
68 : graph_handle(graph_handle), in FunctionData()
|
| D | worker.cc | 95 s = session->graph_mgr->Deregister(request->graph_handle()); in DeregisterGraphAsync()
226 request->graph_handle(), step_id, session.get(), request->exec_opts(), in DoRunGraph()
267 const string& graph_handle = request->graph_handle(); in DoPartialRunGraph() local
313 graph_handle, step_id, session.get(), request->exec_opts(), in DoPartialRunGraph()
|
| D | message_wrappers.cc | 307 const string& InMemoryRunGraphRequest::graph_handle() const { in graph_handle() function in tensorflow::InMemoryRunGraphRequest
401 proto_version_->set_graph_handle(graph_handle()); in ToProto()
435 const string& MutableProtoRunGraphRequest::graph_handle() const { in graph_handle() function in tensorflow::MutableProtoRunGraphRequest
436 return request_.graph_handle(); in graph_handle()
550 const string& ProtoRunGraphRequest::graph_handle() const { in graph_handle() function in tensorflow::ProtoRunGraphRequest
551 return request_->graph_handle(); in graph_handle()
|
| D | message_wrappers.h | 260 virtual const string& graph_handle() const = 0;
329 const string& graph_handle() const override;
387 const string& graph_handle() const override;
428 const string& graph_handle() const override;
|
| D | cluster_function_library_runtime.cc | 158 function_data_.push_back(FunctionData(resp.graph_handle(), options.target, wi, in Instantiate()
187 req->set_graph_handle(function_data->graph_handle); in Run()
|
| D | master_session.cc | 264 string graph_handle; member
488 partitions_[i].graph_handle = c->resp.graph_handle(); in DoRegisterPartitions()
632 c->req->set_graph_handle(part.graph_handle); in RunPartitionsHelper()
1045 if (!part.graph_handle.empty()) { in DeregisterPartitions()
1049 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 | operation_jni.cc | 88 JNIEnv* env, jclass clazz, jlong graph_handle, jlong op_handle, in Java_org_tensorflow_Operation_shape() argument
90 TF_Graph* graph = requireGraphHandle(env, graph_handle); in Java_org_tensorflow_Operation_shape()
140 jlong graph_handle, in Java_org_tensorflow_Operation_dtype() argument
143 TF_Graph* graph = requireGraphHandle(env, graph_handle); in Java_org_tensorflow_Operation_dtype()
|
| D | operation_builder_jni.cc | 55 JNIEnv* env, jclass clazz, jlong graph_handle, jstring type, jstring name) { in Java_org_tensorflow_OperationBuilder_allocate() argument
56 if (graph_handle == 0) { in Java_org_tensorflow_OperationBuilder_allocate()
61 TF_Graph* graph = reinterpret_cast<TF_Graph*>(graph_handle); in Java_org_tensorflow_OperationBuilder_allocate()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 136 // If the registration succeeds, returns an opaque graph_handle to
137 // the master. The master calls RunGraph with graph_handle to
139 string graph_handle = 1; field
146 // The master deregisters the given graph_handle when the graph is no
150 // The worker deregisters a graph_handle automatically according to on
163 // REQUIRED: graph_handle must be returned by a RegisterGraph call
165 string graph_handle = 1; field
196 // The worker executes all subgraphs registered under graph_handle.
216 // search for the graph_handle.
222 // REQUIRED: graph_handle must be returned by a RegisterGraph call
[all …]
|