5
5
6
6
namespace onnxruntime {
7
7
namespace js {
8
- #define REGISTER_RESIZE_VERSIONED_10_10_KERNEL (domain ) \
9
- ONNX_OPERATOR_VERSIONED_KERNEL_EX ( \
10
- Resize, \
11
- domain, \
12
- 10 , 10 , \
13
- kJsExecutionProvider , \
14
- (*KernelDefBuilder::Create ()) \
15
- .InputMemoryType(OrtMemTypeCPUInput, 1 ) \
16
- .TypeConstraint(" T" , DataTypeImpl::GetTensorType< float > ()), \
8
+ #define REGISTER_RESIZE_VERSIONED_10_10_KERNEL (domain ) \
9
+ ONNX_OPERATOR_VERSIONED_KERNEL_EX ( \
10
+ Resize, \
11
+ domain, \
12
+ 10 , 10 , \
13
+ kJsExecutionProvider , \
14
+ (*KernelDefBuilder::Create ()) \
15
+ .InputMemoryType(OrtMemTypeCPUInput, 1 ) \
16
+ .TypeConstraint(" T" , JsepSupportedFloatTypes ()), \
17
17
Resize);
18
18
19
19
#define REGISTER_RESIZE_VERSIONED_KERNEL (domain, sinceVersion, endVerion ) \
@@ -26,22 +26,22 @@ namespace js {
26
26
.InputMemoryType(OrtMemTypeCPUInput, 1 ) \
27
27
.InputMemoryType(OrtMemTypeCPUInput, 2 ) \
28
28
.InputMemoryType(OrtMemTypeCPUInput, 3 ) \
29
- .TypeConstraint(" T1" , DataTypeImpl::GetTensorType< float > ()) \
30
- .TypeConstraint(" T2" , DataTypeImpl::GetTensorType< float > ()), \
29
+ .TypeConstraint(" T1" , JsepSupportedFloatTypes ()) \
30
+ .TypeConstraint(" T2" , JsepSupportedFloatTypes ()), \
31
31
Resize);
32
32
33
- #define REGISTER_RESIZE_KERNEL (domain, sinceVersion ) \
34
- ONNX_OPERATOR_KERNEL_EX ( \
35
- Resize, \
36
- domain, \
37
- sinceVersion, \
38
- kJsExecutionProvider , \
39
- (*KernelDefBuilder::Create ()) \
40
- .InputMemoryType(OrtMemTypeCPUInput, 1 ) \
41
- .InputMemoryType(OrtMemTypeCPUInput, 2 ) \
42
- .InputMemoryType(OrtMemTypeCPUInput, 3 ) \
43
- .TypeConstraint(" T1" , DataTypeImpl::GetTensorType< float > ()) \
44
- .TypeConstraint(" T2" , DataTypeImpl::GetTensorType< float > ()), \
33
+ #define REGISTER_RESIZE_KERNEL (domain, sinceVersion ) \
34
+ ONNX_OPERATOR_KERNEL_EX ( \
35
+ Resize, \
36
+ domain, \
37
+ sinceVersion, \
38
+ kJsExecutionProvider , \
39
+ (*KernelDefBuilder::Create ()) \
40
+ .InputMemoryType(OrtMemTypeCPUInput, 1 ) \
41
+ .InputMemoryType(OrtMemTypeCPUInput, 2 ) \
42
+ .InputMemoryType(OrtMemTypeCPUInput, 3 ) \
43
+ .TypeConstraint(" T1" , JsepSupportedFloatTypes ()) \
44
+ .TypeConstraint(" T2" , JsepSupportedFloatTypes ()), \
45
45
Resize);
46
46
47
47
#define REGISTER_RESIZE_KERNEL_DOMAIN (domain ) \
0 commit comments