Searched refs:c_to_cpp_device_map_ (Results 1 – 2 of 2) sorted by relevance
281 auto it = c_to_cpp_device_map_.find(c_device); in GetCppDevice()282 CHECK(it != c_to_cpp_device_map_.end()); in GetCppDevice()293 absl::flat_hash_map<PJRT_Device*, PjRtCApiDevice*> c_to_cpp_device_map_; variable
75 c_to_cpp_device_map_.reserve(n); in InitDevices()84 c_to_cpp_device_map_[device] = cpp_device.get(); in InitDevices()