1 #pragma once 2 3 // See NOTE: [Tensor vs. TensorBase] 4 namespace at { 5 class TensorBase; 6 } 7 8 namespace at::native { 9 10 TORCH_API bool cudnn_is_acceptable(const TensorBase& self); 11 12 } // namespace at::native 13