Create LossCTCKernels.h

intel · Sep 19, 2024 · 30904df · 30904df
1 parent db3d203
commit 30904df
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/ATen/native/xpu/sycl/LossCTCKernels.h b/src/ATen/native/xpu/sycl/LossCTCKernels.h
@@ -0,0 +1,14 @@
+#pragma once
+#include <ATen/native/TensorIterator.h>
+
+namespace at::native::xpu {
+
+TORCH_XPU_API std::tuple<Tensor, Tensor> ctc_loss_kernel(
+    const Tensor& log_probs,
+    const Tensor& targets,
+    IntArrayRef input_lengths,
+    IntArrayRef target_lengths,
+    int64_t BLANK,
+    bool zero_infinity);
+
+} // namespace at::native::xpu