Split with n splits=1 should be optimized away
Split with n splits=1 should be optimized away: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). Split with n splits=1 should be optimized away When I use theano.tensor.split with n splits=1 (this comes up in generic code where the number of splits is computed from a…
The task
Split with n_splits=1 should be optimized away When I use theano.tensor.split with n_splits=1 (this comes up in generic code where the number of splits is computed from a parameter that can be 1), the resulting graph still contains a full Split Apply node, even though…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.