fix InputLayer gpu mode when output cannot handle is2DSquareReshaped Tensors
fix InputLayer gpu mode when output cannot handle is2DSquareReshaped Tensors

transcranial/keras-js,transcranial/keras-js,transcranial/keras-js