error: 'ttnn.conv2d' op Bias must only have data on the final dimenstion in vgg

[kamalrajkannan78]

Description

• error: 'ttnn.conv2d' op Bias must only have data on the final dimenstion in vgg variants

Reproduce

git checkout kkannan/vgg_conv2d_issue
git submodule update --recursive
# to run model specific sanity
pytest forge/test/model_demos/high_prio/cnn/pytorch/test_vgg.py -k "11" -svv
# to run op specific sanity
pytest forge/test/mlir/test_ops.py -k "issue" -svv 

Observed Behaviour

loc("conv2d_0.dc.conv2d.3"("forward":4294967295:25)): error: 'ttnn.conv2d' op Bias must only have data on the final dimenstion

Logs

• vgg11_model_specific_sanity.log

• vgg11_op_speicific_sanity.log

• vgg11.log

[Kamalrajkannan]

linked to #618