14 lines
242 B
C
14 lines
242 B
C
|
#pragma once
|
||
|
|
#include <c10/macros/Export.h>
|
||
|
|
#include <limits>
|
||
|
|
|
||
|
|
namespace at {
|
||
|
|
class TensorBase;
|
||
|
|
}
|
||
|
|
|
||
|
|
namespace at::native {
|
||
|
|
|
||
|
|
TORCH_API bool canUse32BitIndexMath(const at::TensorBase &t, int64_t max_elem=std::numeric_limits<int32_t>::max());
|
||
|
|
|
||
|
|
}
|