../pruned_transducer_stateless2/encoder_interface.py