| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_data_test.cc | 63 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TEST() local
102 eager_context->HostCPU()->resource_manager(), eager_context, in TEST()
194 EXPECT_EQ(eager_context->GetFunctionDef("main"), nullptr); in TEST()
195 ASSERT_NE(eager_context->GetFunctionDef("add_subgraph"), nullptr); in TEST()
196 ASSERT_NE(eager_context->GetFunctionDef("mul_subgraph"), nullptr); in TEST()
198 *(eager_context->GetFunctionDef("add_subgraph")), add_fdef)); in TEST()
200 *(eager_context->GetFunctionDef("mul_subgraph")), mul_fdef)); in TEST()
202 eager_context->Unref(); in TEST()
208 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TEST() local
233 eager_context->HostCPU()->resource_manager(), eager_context, in TEST()
[all …]
|
| D | kernel.cc | 244 tensorflow::EagerContext* eager_context) { in BuildOpKernelRunner() argument
253 *eager_context->pflr(), in BuildOpKernelRunner()
254 eager_context->local_device_mgr()->HostCPU())); in BuildOpKernelRunner()
393 tensorflow::EagerContext* eager_context; member
447 op_data_->eager_context = flex_delegate_data->GetEagerContext(); in Init()
494 status = node_data.BuildOpKernelRunner(op_data_->eager_context); in Init()
546 context, op_data_->eager_context != nullptr, in Prepare()
728 auto& eager_context = *op_data_->eager_context; in Eval() local
733 run_state.params.step_container = eager_context.StepContainer(); in Eval()
734 auto* device = eager_context.local_device_mgr()->HostCPU(); in Eval()
[all …]
|
| D | delegate_data.cc | 160 tensorflow::EagerContext* eager_context, TfLiteDelegate* flex_delegate) { in RegisterFunctionDefForSubgraphs() argument
187 TF_RETURN_IF_ERROR(eager_context->AddFunctionDef(fdef)); in RegisterFunctionDefForSubgraphs()
|
| D | delegate_data.h | 102 tensorflow::EagerContext* eager_context, TfLiteDelegate* flex_delegate);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 94 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
201 eager_context->RemoteMgr()->AddOperationOutput(retvals[i], op_id, in AddOpRetvalsToResponse()
482 EagerContext* eager_context = context->Context(); in RunComponentFunction() local
483 EagerExecutor* eager_executor = &eager_context->Executor(); in RunComponentFunction()
485 EagerOperation* op = new EagerOperation(eager_context); in RunComponentFunction()
487 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 | 41 RemoteExecuteNode(EagerContext* eager_context, in RemoteExecuteNode() argument
49 eager_context_(eager_context), in RemoteExecuteNode()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/util/ |
| D | fallback_test_util.cc | 38 auto* eager_context = expected_eager_context.get(); in CreateFallbackTestExecutionContext() local
39 assert(eager_context); in CreateFallbackTestExecutionContext()
48 tensorflow::Status status = eager_context->AddFunctionDef(dummy_function_def); in CreateFallbackTestExecutionContext()
55 &request_context_builder, eager_context->local_device_mgr(), in CreateFallbackTestExecutionContext()
56 eager_context->pflr(), user_intra_op_threadpool); in CreateFallbackTestExecutionContext()
|
| /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/core/runtime_fallback/runtime/ |
| D | conversion_function.cc | 148 auto *eager_context = expected_eager_context.get(); in TransferRuntimeFallbackToAnotherDevice() local
152 Status s = eager_context->FindDeviceFromName(dst.name().data(), &tf_device); in TransferRuntimeFallbackToAnotherDevice()
159 s = EagerCopyToDevice(th, eager_context, &eager_context->Executor(), in TransferRuntimeFallbackToAnotherDevice()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/ |
| D | op_handler_selector.cc | 33 EagerContext* eager_context, in EagerOpHandlerSelector() argument
37 eager_context_(eager_context), in EagerOpHandlerSelector()
|
| D | op_handler_selector.h | 44 EagerOpHandlerSelector(CoreRuntime* core_runtime, EagerContext* eager_context,
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | utils.cc | 43 const char* device_name, tensorflow::EagerContext* eager_context) { in ConvertTfDeviceNameToTfrt() argument
49 eager_context->FindDeviceFromName(device_name, &tf_device); in ConvertTfDeviceNameToTfrt()
|
| D | utils.h | 49 const char* device_name, tensorflow::EagerContext* eager_context);
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_execute_compat.cc | 138 tensorflow::EagerContext* eager_context, in SetUpKernelFallbackCompatRequestContext() argument
154 GetDefaultRunner(), eager_context->local_device_mgr(), step_id, in SetUpKernelFallbackCompatRequestContext()
156 eager_context->StepContainer()}, in SetUpKernelFallbackCompatRequestContext()
157 eager_context->GetCollectiveExecutorHandle(), in SetUpKernelFallbackCompatRequestContext()
159 eager_context->RendezvousCreator()(step_id)), in SetUpKernelFallbackCompatRequestContext()
161 model_metadata, eager_context->pflr()); in SetUpKernelFallbackCompatRequestContext()
164 eager_context->LogDevicePlacement()); in SetUpKernelFallbackCompatRequestContext()
|
| D | kernel_fallback_execute_compat.h | 61 tensorflow::EagerContext* eager_context,
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/runtime_fallback/ |
| D | runtime_fallback_executor.cc | 110 auto* eager_context = expected_eager_context.get(); in RuntimeFallbackExecutor() local
114 &builder, /*runner_table=*/nullptr, eager_context, intra_op_.get()); in RuntimeFallbackExecutor()
|
| /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 | 209 from tensorflow.python.eager import context as eager_context unknown
1433 if eager_context.executing_eagerly():
1499 if eager_context.executing_eagerly():
3770 if eager_context.executing_eagerly():
3788 if eager_context.executing_eagerly():
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 145 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TFE_NewContext() local
156 eager_context->SetDistributedManager( in TFE_NewContext()
158 eager_context)); in TFE_NewContext()
160 return tensorflow::wrap(eager_context); in TFE_NewContext()
|
| /external/tensorflow/tensorflow/python/tools/api/generator/ |
| D | api_init_files.bzl | 15 "__internal__/eager_context/__init__.py",
|