Searched defs:tpu_model_resource (Results 1 – 5 of 5) sorted by relevance
59 auto tpu_model_resource = std::make_unique<tfrt::tpu::TpuModelResource>(); in TEST_F() local102 auto tpu_model_resource = std::make_unique<tfrt::tpu::TpuModelResource>(); in TEST_F() local
277 tfrt::tpu::TpuModelResource* tpu_model_resource, in CreateResourceContext()293 tfrt::tpu::TpuModelResource* tpu_model_resource, in Create()
114 tfrt::tpu::TpuModelResource* tpu_model_resource, in GraphExecutor()
27 …ddTpuResources(class ResourceContext* resource_context, tpu::TpuModelResource* tpu_model_resource)… in AddTpuResources()
564 auto tpu_model_resource = std::make_unique<tfrt::tpu::TpuModelResource>(); in LoadSavedModel() local605 std::unique_ptr<tfrt::tpu::TpuModelResource> tpu_model_resource, in SavedModelImpl()