Searched defs:PmapCacheEntry (Results 1 – 1 of 1) sorted by relevance
168 struct PmapCacheEntry { struct169 std::shared_ptr<xla::PyExecutable> executable;171 py::object py_devices; // To pass back to Python.172 std::vector<xla::PjRtDevice*> devices;173 std::vector<InputSpec> input_specs;174 xla::PyTreeDef out_pytree_def;176 std::vector<ResultSpec> out_result_specs;183 absl::Notification compilation_complete;185 bool fall_back_to_python = false;