From f98f0f87178c8848249eb2a77549243bb35480de Mon Sep 17 00:00:00 2001 From: Yulong Wang Date: Fri, 14 Feb 2020 14:13:32 -0800 Subject: [PATCH] fix type constraint name of Resize (#3019) --- .../core/providers/cpu/tensor/resize.cc | 6 +-- .../core/providers/cuda/tensor/resize.cc | 44 +++++++++---------- 2 files changed, 25 insertions(+), 25 deletions(-) diff --git a/onnxruntime/core/providers/cpu/tensor/resize.cc b/onnxruntime/core/providers/cpu/tensor/resize.cc index 4e8ff2f8ab..0d000108e1 100644 --- a/onnxruntime/core/providers/cpu/tensor/resize.cc +++ b/onnxruntime/core/providers/cpu/tensor/resize.cc @@ -33,21 +33,21 @@ ONNX_CPU_OPERATOR_TYPED_KERNEL( Resize, 11, float, - KernelDefBuilder().TypeConstraint("T", DataTypeImpl::GetTensorType()), + KernelDefBuilder().TypeConstraint("T1", DataTypeImpl::GetTensorType()), Resize); ONNX_CPU_OPERATOR_TYPED_KERNEL( Resize, 11, int32_t, - KernelDefBuilder().TypeConstraint("T", DataTypeImpl::GetTensorType()), + KernelDefBuilder().TypeConstraint("T1", DataTypeImpl::GetTensorType()), Resize); ONNX_CPU_OPERATOR_TYPED_KERNEL( Resize, 11, uint8_t, - KernelDefBuilder().TypeConstraint("T", DataTypeImpl::GetTensorType()), + KernelDefBuilder().TypeConstraint("T1", DataTypeImpl::GetTensorType()), Resize); } // namespace onnxruntime diff --git a/onnxruntime/core/providers/cuda/tensor/resize.cc b/onnxruntime/core/providers/cuda/tensor/resize.cc index 4940c872eb..a268ef1052 100644 --- a/onnxruntime/core/providers/cuda/tensor/resize.cc +++ b/onnxruntime/core/providers/cuda/tensor/resize.cc @@ -5,28 +5,28 @@ namespace onnxruntime { namespace cuda { -#define REGISTER_KERNEL_TYPED(T) \ - ONNX_OPERATOR_VERSIONED_TYPED_KERNEL_EX( \ - Resize, \ - kOnnxDomain, \ - 10, 10, \ - T, \ - kCudaExecutionProvider, \ - KernelDefBuilder() \ - .InputMemoryType(1) \ - .TypeConstraint("T", DataTypeImpl::GetTensorType()), \ - Resize); \ - ONNX_OPERATOR_TYPED_KERNEL_EX( \ - Resize, \ - kOnnxDomain, \ - 11, \ - T, \ - kCudaExecutionProvider, \ - KernelDefBuilder() \ - .InputMemoryType(1) \ - .InputMemoryType(2) \ - .InputMemoryType(3) \ - .TypeConstraint("T", DataTypeImpl::GetTensorType()), \ +#define REGISTER_KERNEL_TYPED(T) \ + ONNX_OPERATOR_VERSIONED_TYPED_KERNEL_EX( \ + Resize, \ + kOnnxDomain, \ + 10, 10, \ + T, \ + kCudaExecutionProvider, \ + KernelDefBuilder() \ + .InputMemoryType(1) \ + .TypeConstraint("T", DataTypeImpl::GetTensorType()), \ + Resize); \ + ONNX_OPERATOR_TYPED_KERNEL_EX( \ + Resize, \ + kOnnxDomain, \ + 11, \ + T, \ + kCudaExecutionProvider, \ + KernelDefBuilder() \ + .InputMemoryType(1) \ + .InputMemoryType(2) \ + .InputMemoryType(3) \ + .TypeConstraint("T1", DataTypeImpl::GetTensorType()), \ Resize); REGISTER_KERNEL_TYPED(float)