../pruned_transducer_stateless/model.py