Searched defs:resource_context (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/tfrt/tpu/ |
| D | tpu_resources.h | 21 inline void AddTpuResources(class ResourceContext *resource_context) {} in AddTpuResources()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/ |
| D | op_handler_registry.cc | 32 ResourceContext* resource_context, in RegisterOpHandlers()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/ |
| D | tf_bef_executor_main.cc | 57 -> llvm::Expected<tfrt::ExecutionContext> { in main()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/backends/cpu/ |
| D | cpu_registration.cc | 34 ResourceContext* resource_context, in RegisterCpuOpHandler()
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| D | saved_model.cc | 132 ResourceContext* resource_context, in SetUpRequestContext()
236 tfrt::ResourceContext* resource_context, in RunInitializers()
446 tfrt::ResourceContext* resource_context, in InitSavedModel()
604 auto resource_context = std::make_unique<tfrt::ResourceContext>(); in LoadSavedModel() local
643 std::unique_ptr<tfrt::ResourceContext> resource_context) in SavedModelImpl()
718 tfrt::ResourceContext* resource_context; in Run() local
1124 tfrt::ResourceContext* resource_context) { in RunInternal()
|
| D | saved_model.h | 284 std::unique_ptr<tfrt::ResourceContext> resource_context; member
|
| /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/eager/backends/gpu/ |
| D | gpu_registration.cc | 34 ResourceContext* resource_context, in RegisterGpuOpHandler()
|
| /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_batch_tf_opkernels.cc | 298 HostContext* host_ctx, tfrt::ResourceContext* resource_context, in SetUpRequestContext()
|
| D | runtime_fallback_kernels.cc | 309 tfrt::ResourceContext* resource_context = exec_ctx.resource_context(); in TfdInitEagerContext() local
684 tfrt::ResourceContext* resource_context = exec_ctx.resource_context(); in RuntimeFallbackKernel() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/ |
| D | conversion_function.cc | 195 tfrt::ResourceContext* resource_context = exec_ctx.resource_context(); in ConvertDenseGpuTensorToRuntimeFallbackTensor() local
|