blob: fe6e8395c178e991d5a7a41e69933bafcd028796 [file]
#pragma once
// See NOTE: [Tensor vs. TensorBase]
namespace at {
class TensorBase;
}
namespace at { namespace native {
TORCH_API bool cudnn_is_acceptable(const TensorBase& self);
}} // namespace at::native