Home
last modified time | relevance | path

Searched defs:eager_context (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Deager_service_impl.cc94 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()
H A Dremote_execute_node.h41 RemoteExecuteNode(EagerContext* eager_context, in RemoteExecuteNode()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Ddelegate_data_test.cc63 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TEST() local
208 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TEST() local
H A Dkernel.cc244 tensorflow::EagerContext* eager_context) { in BuildOpKernelRunner()
393 tensorflow::EagerContext* eager_context; member
728 auto& eager_context = *op_data_->eager_context; in Eval() local
H A Ddelegate_data.cc160 tensorflow::EagerContext* eager_context, TfLiteDelegate* flex_delegate) { in RegisterFunctionDefForSubgraphs()
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dpython_tensor_converter_wrapper.cc41 Safe_PyObjectPtr GetAttr_ThreadLocalData(PyObject* eager_context) { in GetAttr_ThreadLocalData()
46 Safe_PyObjectPtr GetAttr_ContextHandle(PyObject* eager_context) { in GetAttr_ContextHandle()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dfallback_test_util.cc38 auto* eager_context = expected_eager_context.get(); in CreateFallbackTestExecutionContext() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/
H A Dop_handler_selector.cc33 EagerContext* eager_context, in EagerOpHandlerSelector()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dutils.cc43 const char* device_name, tensorflow::EagerContext* eager_context) { in ConvertTfDeviceNameToTfrt()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/runtime_fallback/
H A Druntime_fallback_executor.cc110 auto* eager_context = expected_eager_context.get(); in RuntimeFallbackExecutor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Dconversion_function.cc148 auto *eager_context = expected_eager_context.get(); in TransferRuntimeFallbackToAnotherDevice() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dkernel_fallback_execute_compat.cc138 tensorflow::EagerContext* eager_context, in SetUpKernelFallbackCompatRequestContext()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api.cc145 tensorflow::EagerContext* eager_context = new tensorflow::EagerContext( in TFE_NewContext() local
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistribute_lib.py209 from tensorflow.python.eager import context as eager_context unknown