Updated local_alloc_elemwise to remove all allocs when possible and to add assert only when we cannot determine whether the shapes will be correct.
Deprecated experimental.local_alloc_elemwise and enabled local_alloc_elemwise by default.