This commit fixes support for the argmax operation by allowing fp8/bf16 input operands with an int64 output type in the profile compilance such that it aligns with the spec.
Multi-Level Intermediate Representation
See https://mlir.llvm.org/ for more information.