[Bug Report] ...

[asandhupatlaTT]

Describe the bug
Wrong pcc for argmax

To Reproduce
run any argmax test case. For example:
pytest tests/tt_eager/python_api_testing/sweep_tests/pytests/tt_dnn/test_argmax_int.py::TestArgmax::test_argmax[memconfig0--1-input_shapes1]

Expected behavior
Correct PCC

Additional context
Related to #16479

[asandhupatlaTT]

Adding @TT-BrianLiu since i disabled one of his test case.
Adding @bbradelTT for visibility

Similar issue was seen: #11550 but issue was gone (despite having wrong argmax output dimensions)

i fixed the wrong dim issue at #16479 but i get pcc failures. So, i had to disable few test cases.

Snapshot of failing test case:

Input is torch.Size([1, 1, 10, 20]) & output tensor dim is torch.Size([1, 1, 1, 10]) for dim=-1 . Here, PCC is good & test case passes :

Where as, after my patch which fixes the wrong dims of output, i get wrong PCC but correct output tensor shape