TensorProperties.h 210 B

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