Searched defs:DispatchStubImpl (Results 1 – 1 of 1) sorted by relevance
94 struct TORCH_API DispatchStubImpl { struct178 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;[all …]