Searched refs:eager_context (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 96 EagerContext* eager_context, in GetEagerOperationAndNumRetvals() argument
120 eager_context->RemoteMgr()->DeserializeRemoteTensorHandle( in GetEagerOperationAndNumRetvals()
130 nullptr, eager_context); in GetEagerOperationAndNumRetvals()
144 return GetNumRetvals(eager_context, operation.name(), operation.attrs(), in GetEagerOperationAndNumRetvals()
173 EagerContext* eager_context, int op_id, int num_retvals, in AddOpRetvalsToResponse() argument
196 eager_context->RemoteMgr()->AddOperationOutput(retvals[i], op_id, in AddOpRetvalsToResponse()
448 EagerContext* eager_context = context->Context(); in RunComponentFunction() local
449 EagerExecutor* eager_executor = &eager_context->Executor(); in RunComponentFunction()
451 EagerOperation* op = new EagerOperation(eager_context); in RunComponentFunction()
453 s = GetEagerOperationAndNumRetvals(operation, eager_context, eager_executor, in RunComponentFunction()
[all …]
|
| D | eager_service_impl.h | 212 EagerContext* eager_context, EagerExecutor* eager_executor,
215 EagerContext* eager_context);
217 EagerContext* eager_context);
219 EagerContext* eager_context);
|
| D | remote_execute_node.h | 40 RemoteExecuteNode(EagerContext* eager_context, in RemoteExecuteNode() argument
48 eager_context_(eager_context), in RemoteExecuteNode()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_tensor_converter_wrapper.cc | 41 Safe_PyObjectPtr GetAttr_ThreadLocalData(PyObject* eager_context) { in GetAttr_ThreadLocalData() argument
43 return Safe_PyObjectPtr(PyObject_GetAttr(eager_context, attr)); in GetAttr_ThreadLocalData()
46 Safe_PyObjectPtr GetAttr_ContextHandle(PyObject* eager_context) { in GetAttr_ContextHandle() argument
48 return Safe_PyObjectPtr(PyObject_GetAttr(eager_context, attr)); in GetAttr_ContextHandle()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 248 tensorflow::Status BuildEagerOp(tensorflow::EagerContext* eager_context) { in BuildEagerOp() argument
249 op_.reset(new tensorflow::EagerOperation(eager_context)); in BuildEagerOp()
368 tensorflow::EagerContext* eager_context; member
382 op_data_->eager_context = flex_delegate_data->GetEagerContext(); in Init()
415 status = node_data.BuildEagerOp(op_data_->eager_context); in Init()
445 context, op_data_->eager_context != nullptr, in Prepare()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.__internal__.eager_context.pbtxt | 1 path: "tensorflow.__internal__.eager_context"
|
| D | tensorflow.__internal__.pbtxt | 32 name: "eager_context"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 208 from tensorflow.python.eager import context as eager_context unknown
1384 if eager_context.executing_eagerly():
1445 if eager_context.executing_eagerly():
3541 if eager_context.executing_eagerly():
3559 if eager_context.executing_eagerly():
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 144 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TFE_NewContext() local
151 eager_context->SetDistributedManager( in TFE_NewContext()
153 eager_context)); in TFE_NewContext()
155 return tensorflow::wrap(eager_context); in TFE_NewContext()
|
| /external/tensorflow/tensorflow/python/tools/api/generator/ |
| D | api_init_files.bzl | 14 "__internal__/eager_context/__init__.py",
|