Minor changes to complete implementation of local_advanced_indexing_crossentropy_onehot optimization