Searched defs:addressable_devices (Results 1 – 3 of 3) sorted by relevance
77 std::vector<std::unique_ptr<LocalDeviceState>> addressable_devices; in BuildLocalDeviceStates() local91 absl::Span<std::unique_ptr<LocalDeviceState> const> addressable_devices, in CreateBFCAllocator()148 absl::Span<std::unique_ptr<LocalDeviceState> const> addressable_devices) { in GetGpuDeviceAllocator()
140 absl::Span<PjRtDevice* const> addressable_devices() const override { in addressable_devices() function642 absl::Span<PjRtDevice* const> addressable_devices() const override { in addressable_devices() function
1571 std::vector<PjRtDevice*> addressable_devices, in PjRtStreamExecutorExecutable()2150 std::vector<PjRtDevice*> addressable_devices; in Compile() local