[v1.8.x] Fix Nightly Large Tensor test

[ChaiBapchya]

CP #19194

• fixing batch_norm and layer_norm for large tensors (fixing batch_norm and layer_norm for large tensor nightly test #17805)

Co-authored-by: Rohit Kumar Srivastava srivastava.141@buckeyemail.osu.edu

• Fix nightly large_vector test caused by incorrect with_seed path (Fix nightly large_vector test caused by incorrect with_seed path #18178)

• add back the missing environment function

Co-authored-by: Rohit Kumar Srivastava srivastava.141@osu.edu
Co-authored-by: Rohit Kumar Srivastava srivastava.141@buckeyemail.osu.edu

Description

(Brief description on what this PR is about)

Checklist

Essentials

• PR's title starts with a category (e.g. [BUGFIX], [MODEL], [TUTORIAL], [FEATURE], [DOC], etc)
• Changes are complete (i.e. I finished coding on this PR)
• All changes have test coverage
• Code is well-documented

Changes

• Feature1, tests, (and when applicable, API doc)
• Feature2, tests, (and when applicable, API doc)

Comments

• If this change is a backward incompatible change, why must this change be made.
• Interesting edge cases to note here

[mxnet-bot]

Hey @ChaiBapchya , Thanks for submitting the PR
All tests are already queued to run once. If tests fail, you can trigger one or more tests again with the following commands:

• To trigger all jobs: @mxnet-bot run ci [all]
• To trigger specific jobs: @mxnet-bot run ci [job1, job2]

---

CI supported jobs: [centos-gpu, website, edge, unix-cpu, clang, sanity, miscellaneous, centos-cpu, unix-gpu, windows-cpu, windows-gpu]

---

Note:
Only following 3 categories can trigger CI :PR Author, MXNet Committer, Jenkins Admin.
All CI tests must pass before the PR can be merged.

[ChaiBapchya]

Good to merge once v1.x passes: https://jenkins.mxnet-ci.amazon-ml.com/blue/organizations/jenkins/NightlyTestsForBinaries/detail/v1.x/2/pipeline

[ChaiBapchya]

@samskalicky @josephevans

[ChaiBapchya]

1.x nightly is green. Good to merge this PR in 1.8.x