CanUse32BitIndexMath.h 254 B

12345678910111213
  1. #pragma once
  2. #include <c10/macros/Export.h>
  3. #include <limits>
  4. namespace at {
  5. class TensorBase;
  6. }
  7. namespace at { namespace native {
  8. TORCH_API bool canUse32BitIndexMath(const at::TensorBase &t, int64_t max_elem=std::numeric_limits<int32_t>::max());
  9. }}