TensorProperties.h 198 B

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