Searched defs:eager_context (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 98 EagerContext* eager_context, in GetEagerOperationAndNumRetvals()
175 EagerContext* eager_context, int op_id, int num_retvals, in AddOpRetvalsToResponse()
472 EagerContext* eager_context = context->Context(); in RunComponentFunction() local
536 EagerContext* eager_context, in ExecuteOp()
684 const RegisterFunctionOp& register_function, EagerContext* eager_context) { in RegisterFunction()
699 EagerContext* eager_context) { in SendTensor()
726 const SendPackedHandleOp& send_packed_handle, EagerContext* eager_context) { in SendPackedHandle()
|
| D | remote_execute_node.h | 40 RemoteExecuteNode(EagerContext* eager_context, in RemoteExecuteNode()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_data_test.cc | 62 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TEST() local
206 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TEST() local
|
| D | delegate_data.cc | 154 tensorflow::EagerContext* eager_context) { in RegisterFunctionDefForSubgraphs()
|
| D | kernel.cc | 249 tensorflow::EagerContext* eager_context, in BuildEagerOp()
372 tensorflow::EagerContext* eager_context; member
|
| /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 | 31 const char* device_name, tensorflow::EagerContext* eager_context) { in ConvertTfDeviceNameToTfrt()
|
| /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 | 118 tensorflow::EagerContext* eager_context, in SetUpKernelFallbackCompatRequestContext()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 144 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TFE_NewContext() local
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 210 from tensorflow.python.eager import context as eager_context unknown
|