Searched refs:PjRtLoadedExecutable (Results 1 – 18 of 18) sorted by relevance
161 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> Compile( in Compile()170 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> Compile(174 const PjRtLoadedExecutable& executable) const override;177 const PjRtLoadedExecutable& executable) const override;179 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> DeserializeExecutable(270 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> WrapExecutable(271 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> to_wrap);423 class PjRtCApiExecutable : public PjRtLoadedExecutable {426 std::unique_ptr<PjRtLoadedExecutable> wrapped);492 PjRtLoadedExecutable* wrapped() const;[all …]
154 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> Compile(156 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> Compile(160 const PjRtLoadedExecutable& executable) const override;163 const PjRtLoadedExecutable& executable) const override { in SerializeExecutable()168 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> DeserializeExecutable( in DeserializeExecutable()468 class TfrtCpuExecutable final : public PjRtLoadedExecutable {515 using PjRtLoadedExecutable::Execute;522 using PjRtLoadedExecutable::ExecuteSharded;529 using PjRtLoadedExecutable::ExecutePortable;
377 class PjRtLoadedExecutable; variable499 virtual StatusOr<std::unique_ptr<PjRtLoadedExecutable>> Compile(503 virtual StatusOr<std::unique_ptr<PjRtLoadedExecutable>> Compile(508 const PjRtLoadedExecutable& executable) const = 0;514 const PjRtLoadedExecutable& executable) const = 0;519 virtual StatusOr<std::unique_ptr<PjRtLoadedExecutable>> DeserializeExecutable(528 virtual StatusOr<std::unique_ptr<PjRtLoadedExecutable>>539 virtual StatusOr<std::unique_ptr<PjRtLoadedExecutable>> Load( in Load()1106 class PjRtLoadedExecutable : public PjRtExecutable {1108 virtual ~PjRtLoadedExecutable() = default;
193 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> Compile(195 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> Compile(199 const PjRtLoadedExecutable& executable) const override { in ExecutableFingerprint()204 const PjRtLoadedExecutable& executable) const override { in SerializeExecutable()209 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> DeserializeExecutable( in DeserializeExecutable()333 std::vector<PjRtLoadedExecutable::LogicalDeviceIds>710 class PjRtStreamExecutorExecutable : public PjRtLoadedExecutable {759 using PjRtLoadedExecutable::Execute;766 using PjRtLoadedExecutable::ExecuteSharded;773 using PjRtLoadedExecutable::ExecutePortable;
157 const PjRtLoadedExecutable& executable) const { in ExecutableFingerprint()223 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> PjRtCApiClient::Compile( in Compile()240 std::unique_ptr<PjRtLoadedExecutable> ret = in Compile()246 const PjRtLoadedExecutable& executable) const { in SerializeExecutable()254 StatusOr<std::unique_ptr<PjRtLoadedExecutable>>271 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> PjRtCApiClient::WrapExecutable( in WrapExecutable()272 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> to_wrap) { in WrapExecutable()273 TF_ASSIGN_OR_RETURN(std::unique_ptr<PjRtLoadedExecutable> executable, in WrapExecutable()275 return std::unique_ptr<PjRtLoadedExecutable>( in WrapExecutable()415 PjRtCApiClient* client, std::unique_ptr<PjRtLoadedExecutable> wrapped) in PjRtCApiExecutable()[all …]
87 const PjRtLoadedExecutable& executable) const override;90 const PjRtLoadedExecutable& executable) const override;92 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> DeserializeExecutable(
136 const PjRtLoadedExecutable& executable) const { in ExecutableFingerprint()157 const PjRtLoadedExecutable& executable) const { in SerializeExecutable()171 StatusOr<std::unique_ptr<PjRtLoadedExecutable>>207 return std::unique_ptr<PjRtLoadedExecutable>(std::move(pjrt_executable)); in DeserializeExecutable()
242 const PjRtLoadedExecutable& executable) const { in ExecutableFingerprint()332 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> TfrtCpuClient::Compile( in Compile()352 std::vector<PjRtLoadedExecutable::LogicalDeviceIds> in Compile()366 PjRtLoadedExecutable::LogicalDeviceIds logica_device_ids; in Compile()418 return std::unique_ptr<PjRtLoadedExecutable>(std::move(executable)); in Compile()421 StatusOr<std::unique_ptr<PjRtLoadedExecutable>> TfrtCpuClient::Compile( in Compile()1303 StatusOr<PjRtLoadedExecutable::Result> TfrtCpuExecutable::ExecuteHelper( in ExecuteHelper()
59 std::unique_ptr<PjRtLoadedExecutable> executable, in TEST()
2090 StatusOr<PjRtLoadedExecutable::Result>2426 PjRtLoadedExecutable::LogicalDeviceIds logica_device_ids; in GetExecutableExtras()2447 StatusOr<std::unique_ptr<PjRtLoadedExecutable>>2482 return std::unique_ptr<PjRtLoadedExecutable>(std::move(executable)); in Compile()2485 StatusOr<std::unique_ptr<PjRtLoadedExecutable>>
69 std::unique_ptr<PjRtLoadedExecutable> MakeIncrementProgram( in MakeIncrementProgram()
74 std::unique_ptr<PjRtLoadedExecutable> executable,81 std::shared_ptr<PjRtLoadedExecutable> executable() const { in executable()85 absl::Span<const PjRtLoadedExecutable::LogicalDeviceIds>127 const PjRtLoadedExecutable& pjrt_executable() const { return *executable_; } in pjrt_executable()129 PjRtLoadedExecutable* mutable_pjrt_executable() const { in mutable_pjrt_executable()151 std::shared_ptr<PjRtLoadedExecutable> executable_;
321 std::unique_ptr<PjRtLoadedExecutable> executable; in Compile()339 std::unique_ptr<PjRtLoadedExecutable> executable; in CompileMlir()365 std::unique_ptr<PjRtLoadedExecutable> executable; in DeserializeExecutable()
47 std::unique_ptr<PjRtLoadedExecutable> executable, in PyExecutable()
414 TF_ASSIGN_OR_RETURN(std::unique_ptr<PjRtLoadedExecutable> executable, in SendShutdownOutfeedHeader()
46 TF_ASSIGN_OR_RETURN(std::unique_ptr<PjRtLoadedExecutable> executable, in CompileAndExecute()
54 std::unique_ptr<xla::PjRtLoadedExecutable> executable;
244 PJRT_ASSIGN_OR_RETURN(std::unique_ptr<xla::PjRtLoadedExecutable> executable, in PJRT_Client_Compile()