Searched defs:eager_context (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 94 EagerContext* eager_context, in GetEagerOperationAndNumRetvals()
173 EagerContext* eager_context, int op_id, int num_retvals, in AddOpRetvalsToResponse()
482 EagerContext* eager_context = context->Context(); in RunComponentFunction() local
552 EagerContext* eager_context, in ExecuteOp()
700 const RegisterFunctionOp& register_function, EagerContext* eager_context) { in RegisterFunction()
715 EagerContext* eager_context) { in SendTensor()
742 const SendPackedHandleOp& send_packed_handle, EagerContext* eager_context) { in SendPackedHandle()
|
| D | remote_execute_node.h | 41 RemoteExecuteNode(EagerContext* eager_context, in RemoteExecuteNode()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_data_test.cc | 63 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TEST() local
208 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TEST() local
|
| D | kernel.cc | 244 tensorflow::EagerContext* eager_context) { in BuildOpKernelRunner()
393 tensorflow::EagerContext* eager_context; member
728 auto& eager_context = *op_data_->eager_context; in Eval() local
|
| D | delegate_data.cc | 160 tensorflow::EagerContext* eager_context, TfLiteDelegate* flex_delegate) { in RegisterFunctionDefForSubgraphs()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_tensor_converter_wrapper.cc | 41 Safe_PyObjectPtr GetAttr_ThreadLocalData(PyObject* eager_context) { in GetAttr_ThreadLocalData()
46 Safe_PyObjectPtr GetAttr_ContextHandle(PyObject* eager_context) { in GetAttr_ContextHandle()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/util/ |
| D | fallback_test_util.cc | 38 auto* eager_context = expected_eager_context.get(); in CreateFallbackTestExecutionContext() local
|
| /external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/ |
| D | op_handler_selector.cc | 33 EagerContext* eager_context, in EagerOpHandlerSelector()
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | utils.cc | 43 const char* device_name, tensorflow::EagerContext* eager_context) { in ConvertTfDeviceNameToTfrt()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/runtime_fallback/ |
| D | runtime_fallback_executor.cc | 110 auto* eager_context = expected_eager_context.get(); in RuntimeFallbackExecutor() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | conversion_function.cc | 148 auto *eager_context = expected_eager_context.get(); in TransferRuntimeFallbackToAnotherDevice() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_execute_compat.cc | 138 tensorflow::EagerContext* eager_context, in SetUpKernelFallbackCompatRequestContext()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 145 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TFE_NewContext() local
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 209 from tensorflow.python.eager import context as eager_context unknown
|