make fill_random_weights work for MTIA tbe module #716

Job	Run time
build_artifact (x86, linux.24xlarge, 3.8, 11.8.0)	17m 3s
build_artifact (x86, linux.24xlarge, 3.8, 12.1.1)	21m 18s
build_artifact (x86, linux.24xlarge, 3.9, 11.8.0)	16m 58s
build_artifact (x86, linux.24xlarge, 3.9, 12.1.1)	21m 14s
build_artifact (x86, linux.24xlarge, 3.10, 11.8.0)	17m 34s
build_artifact (x86, linux.24xlarge, 3.10, 12.1.1)	21m 37s
build_artifact (x86, linux.24xlarge, 3.11, 11.8.0)	17m 33s
build_artifact (x86, linux.24xlarge, 3.11, 12.1.1)	21m 40s
build_artifact (x86, linux.24xlarge, 3.12, 11.8.0)	17m 25s
build_artifact (x86, linux.24xlarge, 3.12, 12.1.1)	21m 5s
test_and_publish_artifact (x86, linux.g5.4xlarge.nvidia.gpu, 3.8, 11.8.0, 12.1.1)	18m 30s
test_and_publish_artifact (x86, linux.g5.4xlarge.nvidia.gpu, 3.8, 12.1.1, 12.1.1)	19m 27s
test_and_publish_artifact (x86, linux.g5.4xlarge.nvidia.gpu, 3.9, 11.8.0, 12.1.1)	17m 35s
test_and_publish_artifact (x86, linux.g5.4xlarge.nvidia.gpu, 3.9, 12.1.1, 12.1.1)	18m 30s
test_and_publish_artifact (x86, linux.g5.4xlarge.nvidia.gpu, 3.10, 11.8.0, 12.1.1)	17m 55s
test_and_publish_artifact (x86, linux.g5.4xlarge.nvidia.gpu, 3.10, 12.1.1, 12.1.1)	18m 34s
test_and_publish_artifact (x86, linux.g5.4xlarge.nvidia.gpu, 3.11, 11.8.0, 12.1.1)	17m 55s
test_and_publish_artifact (x86, linux.g5.4xlarge.nvidia.gpu, 3.11, 12.1.1, 12.1.1)	18m 20s
test_and_publish_artifact (x86, linux.g5.4xlarge.nvidia.gpu, 3.12, 11.8.0, 12.1.1)	17m 23s
test_and_publish_artifact (x86, linux.g5.4xlarge.nvidia.gpu, 3.12, 12.1.1, 12.1.1)	18m 7s
	6h 15m 43s

Provide feedback