Searched refs:tpu_model_resource (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/graph_executor/ |
| H A D | graph_executor_test.cc | 59 auto tpu_model_resource = std::make_unique<tfrt::tpu::TpuModelResource>(); in TEST_F() local
62 std::move(options), *fallback_state, tpu_model_resource.get(), graph_def); in TEST_F()
102 auto tpu_model_resource = std::make_unique<tfrt::tpu::TpuModelResource>(); in TEST_F() local
106 tpu_model_resource.get(), graph_def)); in TEST_F()
|
| H A D | graph_executor.h | 75 const Runtime& runtime, tfrt::tpu::TpuModelResource* tpu_model_resource,
109 tfrt::tpu::TpuModelResource* tpu_model_resource,
114 tfrt::tpu::TpuModelResource* tpu_model_resource, in GraphExecutor() argument
119 tpu_model_resource_(tpu_model_resource), in GraphExecutor()
|
| H A D | graph_executor.cc | 277 tfrt::tpu::TpuModelResource* tpu_model_resource, in CreateResourceContext() argument
286 AddTpuResources(resource_context.get(), tpu_model_resource); in CreateResourceContext()
293 tfrt::tpu::TpuModelResource* tpu_model_resource, in Create() argument
309 tpu_model_resource, in Create()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| H A D | saved_model.cc | 564 auto tpu_model_resource = std::make_unique<tfrt::tpu::TpuModelResource>(); in LoadSavedModel() local
566 *options.graph_execution_options.runtime, tpu_model_resource.get(), in LoadSavedModel()
581 tpu_model_resource.get(), in LoadSavedModel()
589 std::move(fallback_state), std::move(tpu_model_resource), in LoadSavedModel()
605 std::unique_ptr<tfrt::tpu::TpuModelResource> tpu_model_resource, in SavedModelImpl() argument
618 tpu_model_resource_(std::move(tpu_model_resource)), in SavedModelImpl()
|
| H A D | saved_model.h | 215 std::unique_ptr<tfrt::tpu::TpuModelResource> tpu_model_resource,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/tpu/ |
| H A D | tpu_resources.h | 27 …TpuResources(class ResourceContext* resource_context, tpu::TpuModelResource* tpu_model_resource) {} in AddTpuResources() argument
|
| H A D | BUILD | 17 "//third_party/tf_runtime_google:tpu_model_resource",
|