[mlir][tosa] Added shape inference for tosa convolution operations
Added shape inference handles cases for convolution operations. This includes conv2d, conv3d, depthwise_conv2d, and transpose_conv2d. With transpose conv we use the specified output shape when possible however will shape propagate if the output shape attribute has dynamic values. Reviewed By: jpienaar Differential Revision: https://reviews.llvm.org/D105645
parent
03cdb522
Please register or sign in to comment