Updated amax_sync test to set const weights based on rank#451

kinjalpatel27 merged 2 commits intomainNVIDIA/TensorRT-Model-Optimizer:mainfrom

kinjal/fix_test_amax_syncNVIDIA/TensorRT-Model-Optimizer:kinjal/fix_test_amax_syncCopy head branch name to clipboard