diff --git a/onnxruntime/core/providers/vsinpu/builders/impl/conv_op_builder.h b/onnxruntime/core/providers/vsinpu/builders/impl/conv_op_builder.h index 1f8f343cf7ca0..f4af4024ae08e 100644 --- a/onnxruntime/core/providers/vsinpu/builders/impl/conv_op_builder.h +++ b/onnxruntime/core/providers/vsinpu/builders/impl/conv_op_builder.h @@ -28,7 +28,7 @@ namespace vsi { namespace npu { class ConvOpBuilder : public BaseOpBuilder { bool IsOpSupported(const onnxruntime::GraphViewer& graph_viewer, - const Node* node) { + const Node* node) const override { auto input_defs = node->InputDefs(); auto shape = vsi::npu::util::GetTensorShape(*input_defs[0]); if (shape.NumDimensions() == 5) { @@ -62,7 +62,7 @@ class ConvOpBuilder : public BaseOpBuilder { helper.Get("strides", is_1d_conv ? std::vector{default_uint} : default_vec); auto dilation = - helper.Get("dilation", is_1d_conv ? std::vector{default_uint} + helper.Get("dilations", is_1d_conv ? std::vector{default_uint} : default_vec); std::shared_ptr op;