saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-18 21:21:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Get onnxruntime/core/providers/cuda/tensor/slice.h from ort_training.

Browse source
This commit is contained in:
Sergii Dymchenko 2020-04-09 17:03:58 -07:00
parent 0e4080f1d6
commit 6bbc80951d
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/providers/cuda/tensor/slice.h
View file

@ -9,8 +9,8 @@
namespace onnxruntime {
namespace cuda {
template <typename Tind, bool dynamic>
class Slice final : public CudaKernel, public SliceBase {
template <bool dynamic>
class Slice : public CudaKernel, public SliceBase {
public:
Slice(const OpKernelInfo& info) : CudaKernel(info), SliceBase(info, dynamic) {}