Searched defs:FusedKernelCUDA (Results 1 – 2 of 2) sorted by relevance
25 struct TORCH_CUDA_CU_API FusedKernelCUDA struct42 at::Backend backend() const override { in backend()51 at::DeviceIndex device_;54 std::vector<char> ptx_;
84 FusedKernelCUDA::FusedKernelCUDA( in FusedKernelCUDA() function in torch::jit::fuser::cuda::FusedKernelCUDA