Searched defs:gpu_global_device_ids (Results 1 – 5 of 5) sorted by relevance
32 std::optional<std::vector<GlobalDeviceId>> gpu_global_device_ids) { in set_gpu_global_device_ids()38 GpuExecutableRunOptions::gpu_global_device_ids() const { in gpu_global_device_ids() function in xla::gpu::GpuExecutableRunOptions
91 const std::vector<GlobalDeviceId>* gpu_global_device_ids; // may be null member
293 std::vector<int64> gpu_global_device_ids; in CreateXcclContext() local
27 std::vector<int64_t> gpu_global_device_ids, in GpuSharedContext()
57 const std::vector<int64_t>& gpu_global_device_ids() const { in gpu_global_device_ids() function