Searched defs:resource_context (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/tfrt/tpu/ |
| D | tpu_resources.h | 27 inline void AddTpuResources(class ResourceContext* resource_context, tpu::TpuModelResource* tpu_mod… in AddTpuResources()
|
| /external/tensorflow/tensorflow/core/tfrt/graph_executor/ |
| D | graph_executor.cc | 85 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()
|
| D | graph_executor.h | 89 std::unique_ptr<tfrt::ResourceContext> resource_context; member
|
| /external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/ |
| D | op_handler_registry.cc | 32 ResourceContext* resource_context, in RegisterOpHandlers()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/backends/cpu/ |
| D | cpu_registration.cc | 34 ResourceContext* resource_context, in RegisterCpuOpHandler()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/ |
| D | tf_bef_executor_main.cc | 61 -> llvm::Expected<tfrt::ExecutionContext> { in main()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/backends/gpu/ |
| D | gpu_registration.cc | 34 ResourceContext* resource_context, in RegisterGpuOpHandler()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/util/ |
| D | fallback_test_util.cc | 26 tfrt::HostContext* host, tfrt::ResourceContext* resource_context, in CreateFallbackTestExecutionContext()
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| D | saved_model.cc | 200 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
|
| D | saved_model.h | 253 std::unique_ptr<tfrt::ResourceContext> resource_context; member
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | kernel_utils.cc | 83 tfrt::ResourceContext* resource_context = exec_ctx.resource_context(); in GetEagerContext() local
|
| D | runtime_fallback_kernels.cc | 311 tfrt::ResourceContext* resource_context = exec_ctx.resource_context(); in TfdInitEagerContext() local
684 tfrt::ResourceContext* resource_context = exec_ctx.resource_context(); in RuntimeFallbackKernel() local
|
| D | runtime_fallback_batch_tf_opkernels.cc | 512 HostContext* host_ctx, tfrt::ResourceContext* resource_context, in SetUpRequestContext()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/ |
| D | conversion_function.cc | 195 tfrt::ResourceContext* resource_context = exec_ctx.resource_context(); in ConvertDenseGpuTensorToRuntimeFallbackTensor() local
|