Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/graph_executor/
H A Dgraph_executor.cc85 tfrt::ResourceContext* resource_context, in SetUpRequestContext()
147 tfrt::ResourceContext* resource_context, const Runtime& runtime, in GraphExecutionRunOnFunction()
279 auto resource_context = std::make_unique<tfrt::ResourceContext>(); in CreateResourceContext() local
526 tfrt::BEFFile* bef_file, tfrt::ResourceContext* resource_context, in InitBef()
H A Dgraph_executor.h89 std::unique_ptr<tfrt::ResourceContext> resource_context; member
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/
H A Dsaved_model.cc200 const Runtime& runtime, tfrt::ResourceContext* resource_context, in RunInitializers()
366 tfrt::ResourceContext* resource_context, in InitSavedModel()
565 auto resource_context = CreateResourceContext( in LoadSavedModel() local
606 std::unique_ptr<tfrt::ResourceContext> resource_context, in SavedModelImpl()
694 tfrt::ResourceContext* resource_context; in Run() local
H A Dsaved_model.h253 std::unique_ptr<tfrt::ResourceContext> resource_context; member
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/tpu/
H A Dtpu_resources.h27 inline void AddTpuResources(class ResourceContext* resource_context, tpu::TpuModelResource* tpu_mod… in AddTpuResources()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/
H A Dop_handler_registry.cc32 ResourceContext* resource_context, in RegisterOpHandlers()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/backends/cpu/
H A Dcpu_registration.cc34 ResourceContext* resource_context, in RegisterCpuOpHandler()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/
H A Dtf_bef_executor_main.cc61 -> llvm::Expected<tfrt::ExecutionContext> { in main()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dfallback_test_util.cc26 tfrt::HostContext* host, tfrt::ResourceContext* resource_context, in CreateFallbackTestExecutionContext()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/backends/gpu/
H A Dgpu_registration.cc34 ResourceContext* resource_context, in RegisterGpuOpHandler()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Dkernel_utils.cc83 tfrt::ResourceContext* resource_context = exec_ctx.resource_context(); in GetEagerContext() local
H A Druntime_fallback_kernels.cc311 tfrt::ResourceContext* resource_context = exec_ctx.resource_context(); in TfdInitEagerContext() local
684 tfrt::ResourceContext* resource_context = exec_ctx.resource_context(); in RuntimeFallbackKernel() local
H A Druntime_fallback_batch_tf_opkernels.cc512 HostContext* host_ctx, tfrt::ResourceContext* resource_context, in SetUpRequestContext()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/
H A Dconversion_function.cc195 tfrt::ResourceContext* resource_context = exec_ctx.resource_context(); in ConvertDenseGpuTensorToRuntimeFallbackTensor() local