#if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION) #pragma once #include namespace at { class TensorBase; } namespace at::native { void launch_fused_mode_kernel( const TensorBase &values, const TensorBase &indices, const TensorBase &self, int64_t slice_size, int64_t slices); void launch_apply_mode_kernel( const TensorBase &values, const TensorBase &indices, const TensorBase &self, int64_t dim, int64_t ndim); } // namespace at::native #else #error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined." #endif // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)