Searched defs:cpp_device (Results 1 – 2 of 2) sorted by relevance
166 xla::PjRtDevice* cpp_device, absl::Span<PJRT_Device* const> device_list) { in FindDeviceWrapper()192 xla::PjRtDevice* cpp_device = cpp_devices[i]; in PopulatePjrtExecutableAddressableDevices() local
81 std::unique_ptr<PjRtCApiDevice>& cpp_device = owned_devices_.emplace_back( in InitDevices() local