Searched refs:pjrt_executable (Results 1 – 7 of 7) sorted by relevance
49 TF_ASSERT_OK_AND_ASSIGN(auto pjrt_executable, in TEST()62 auto result = pjrt_executable->Execute(/*argument_handles=*/{{buffer.get()}}, in TEST()68 result = pjrt_executable->Execute(/*argument_handles=*/{{buffer.get()}}, in TEST()
200 auto pjrt_executable = std::make_unique<PjRtStreamExecutorExecutable>( in DeserializeExecutable() local206 pjrt_executable->SetUpDonation(options.parameter_is_tupled_arguments)); in DeserializeExecutable()207 return std::unique_ptr<PjRtLoadedExecutable>(std::move(pjrt_executable)); in DeserializeExecutable()
160 ":pjrt_executable",226 name = "pjrt_executable",227 hdrs = ["pjrt_executable.h"],244 ":pjrt_executable",
127 const PjRtLoadedExecutable& pjrt_executable() const { return *executable_; } in pjrt_executable() function
511 auto pjrt_executable = cache_entry.executable->mutable_pjrt_executable(); in Call() local512 TF_ASSIGN_OR_RETURN(output_buffers, pjrt_executable->Execute( in Call()
359 return pjrt_client_->SerializeExecutable(executable.pjrt_executable()); in SerializeExecutable()
785 cache_entry->executable->pjrt_executable().addressable_devices().size(); in PopulateCacheEntry()