Home
last modified time | relevance | path

Searched defs:eager_context_resource (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dfallback_test_util.cc32 auto* eager_context_resource = in CreateFallbackTestExecutionContext() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dtfrt_context.cc40 tensorflow::tfd::EagerContextResource* eager_context_resource = in TfrtContext() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Dkernel_utils.cc84 tensorflow::tfd::EagerContextResource* eager_context_resource = in GetEagerContext() local
H A Dconversion_function.cc138 auto eager_context_resource = in TransferRuntimeFallbackToAnotherDevice() local
H A Druntime_fallback_kernels.cc312 tensorflow::tfd::EagerContextResource* eager_context_resource = in TfdInitEagerContext() local
685 EagerContextResource* eager_context_resource = in RuntimeFallbackKernel() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/runtime_fallback/
H A Druntime_fallback_executor.cc105 auto* eager_context_resource = in RuntimeFallbackExecutor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/
H A Dconversion_function.cc196 tensorflow::tfd::EagerContextResource* eager_context_resource = in ConvertDenseGpuTensorToRuntimeFallbackTensor() local