Lines Matching defs:DispatchStubImpl

94 struct TORCH_API DispatchStubImpl {  struct
178     std::atomic<void*> cpu_dispatch_ptr;
179     void* cuda_dispatch_ptr;
180     void* hip_dispatch_ptr;
181     void* mps_dispatch_ptr;
182     void* mtia_dispatch_ptr;
184     void* xpu_dispatch_ptr;
186     void* privateuse1_dispatch_ptr;
189     void* cuda_dispatch_ptr = nullptr;
190     void* hip_dispatch_ptr = nullptr;
191     void* mps_dispatch_ptr = nullptr;
192     void* mtia_dispatch_ptr = nullptr;
194     void* xpu_dispatch_ptr = nullptr;
196     void* privateuse1_dispatch_ptr = nullptr;