Searched defs:PmapCacheEntry (Results 1 – 1 of 1) sorted by relevance
101 struct PmapCacheEntry { struct105 std::shared_ptr<xla::PyExecutable> executable;107 py::object backend;110 py::function handle_args;113 py::function out_handler;114 xla::PyTreeDef out_pytree_def;121 absl::Notification compilation_complete;122 absl::optional<xla::Status> compilation_error = absl::nullopt;124 bool fall_back_to_python = false;