Adding tests for specific shape-related errors in BatchNorm and LayerNorm (#462)
Adding tests for specific shape-related errors in BatchNorm and LayerNorm (#462)

* Added tests for specific shape-related BatchNorm and LayerNorm errors.

* Update Tests/TensorFlowTests/LayerTests.swift

Co-Authored-By: Anthony Platanios <91357c58cb54a038043ea4805bdef6ddc972df32@gmail.com>

tensorflow/swift-apis,tensorflow/swift-apis,tensorflow/swift-apis,tensorflow/swift-apis,tensorflow/swift-apis