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.
Incorporated reviewer comments.
Created tests.
Removed unnecessary assert.
Added documentation for config.experimental.local_alloc_elemwise_assert
Incorporated review comments.