Searched defs:addressable_devices (Results 1 – 9 of 9) sorted by relevance
60 const std::map<int, std::unique_ptr<LocalDeviceState>>& addressable_devices, in CreateCudaAsyncAllocator()192 std::map<int, std::unique_ptr<LocalDeviceState>> addressable_devices; in BuildLocalDeviceStates() local207 const std::map<int, std::unique_ptr<LocalDeviceState>>& addressable_devices, in CreateBFCAllocator()269 const std::map<int, std::unique_ptr<LocalDeviceState>>& in GetGpuDeviceAllocator()
164 absl::Span<PjRtDevice* const> addressable_devices() const override { in addressable_devices() function335 std::vector<PjRtDevice*> addressable_devices; member751 absl::Span<PjRtDevice* const> addressable_devices() const override { in addressable_devices() function
130 absl::Span<PjRtDevice* const> addressable_devices() const override { in addressable_devices() function505 absl::Span<PjRtDevice* const> addressable_devices() const override { in addressable_devices() function
1766 std::vector<PjRtDevice*> addressable_devices, in PjRtStreamExecutorExecutable()2395 std::vector<PjRtDevice*>& addressable_devices = extras.addressable_devices; in GetExecutableExtras() local2458 std::vector<PjRtDevice*>& addressable_devices = extras.addressable_devices; in Compile() local
459 absl::Span<PjRtDevice* const> addressable_devices() const override { in addressable_devices() function
354 std::vector<PjRtDevice*> addressable_devices; in Compile() local1165 std::vector<PjRtDevice*> addressable_devices, TfrtCpuClient* client) in TfrtCpuExecutable()
118 absl::Span<PjRtDevice* const> PjRtCApiClient::addressable_devices() const { in addressable_devices() function in xla::PjRtCApiClient
39 std::vector<PJRT_Device*> addressable_devices; member59 std::vector<PJRT_Device*> addressable_devices; member
281 PJRT_Device** addressable_devices; // out member530 PJRT_Device** addressable_devices; // out member