Searched refs:DevicePutFunc (Results 1 – 1 of 1) sorted by relevance
567 using DevicePutFunc = std::function<xla::StatusOr<DevicePutResult>( typedef744 static const absl::flat_hash_map<PyObject*, DevicePutFunc>* const handlers = in DevicePut()746 auto p = new absl::flat_hash_map<PyObject*, DevicePutFunc>(); in DevicePut()