| 12345678910111213141516171819 |
- #pragma once
- #include <algorithm>
- #include <cstddef>
- #include <exception>
- #include <c10/util/Exception.h>
- #define INTRA_OP_PARALLEL
- namespace at::internal {
- TORCH_API void invoke_parallel(
- const int64_t begin,
- const int64_t end,
- const int64_t grain_size,
- const std::function<void(int64_t, int64_t)>& f);
- } // namespace at::internal
|
