diff options
| author | Kevin Cheng <kevin.cheng@arm.com> | 2021-08-31 16:04:40 -0700 |
|---|---|---|
| committer | Kevin Cheng <kevin.cheng@arm.com> | 2021-08-31 17:23:15 -0700 |
| commit | 79a4199be5b87eb6220b6688b02f5f6cbebb6131 (patch) | |
| tree | 01ac992667773b5b3506a4a3fed1a3fad85088ab /include/operator.def | |
| parent | ab905ec865a9f4889e6818ce1c4bc934cff2070d (diff) | |
| download | serialization_lib-79a4199be5b87eb6220b6688b02f5f6cbebb6131.tar.gz | |
Adding Conv attribute to Conv3d.
- Rename attribute: Pool2d, Conv2d, TransposeConv2d -> Pool, Conv, TransposeConv
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: If47dfd1ffc57843018d81d674e4386212fd8d4b4
Diffstat (limited to 'include/operator.def')
| -rw-r--r-- | include/operator.def | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/include/operator.def b/include/operator.def index 83155d8..386e72a 100644 --- a/include/operator.def +++ b/include/operator.def @@ -28,14 +28,14 @@ /* tensor operators */ DEF_OPERATOR(argmax, ARGMAX, ArgMax, Axis, None) -DEF_OPERATOR(avg_pool2d, AVG_POOL2D, AvgPool2d, Pool2d, Unary) -DEF_OPERATOR(conv2d, CONV2D, Conv2d, Conv2d, Conv) -DEF_OPERATOR(conv3d, CONV3D, Conv3d, None, None) -DEF_OPERATOR(depthwise_conv2d, DEPTHWISE_CONV2D, DepthwiseConv2d, Conv2d, Conv) +DEF_OPERATOR(avg_pool2d, AVG_POOL2D, AvgPool2d, Pool, Unary) +DEF_OPERATOR(conv2d, CONV2D, Conv2d, Conv, Conv) +DEF_OPERATOR(conv3d, CONV3D, Conv3d, Conv, Conv) +DEF_OPERATOR(depthwise_conv2d, DEPTHWISE_CONV2D, DepthwiseConv2d, Conv, Conv) DEF_OPERATOR(fully_connected, FULLY_CONNECTED, FullyConnected, None, Conv) DEF_OPERATOR(matmul, MATMUL, MatMul, None, MatMul) -DEF_OPERATOR(max_pool2d, MAX_POOL2D, MaxPool2d, Pool2d, None) -DEF_OPERATOR(transpose_conv2d, TRANSPOSE_CONV2D, TransposeConv2d, TransposeConv2d, Conv) +DEF_OPERATOR(max_pool2d, MAX_POOL2D, MaxPool2d, Pool, None) +DEF_OPERATOR(transpose_conv2d, TRANSPOSE_CONV2D, TransposeConv2d, TransposeConv, Conv) /* activation */ DEF_OPERATOR(clamp, CLAMP, Clamp, Clamp, None) |