def _ForwardPass(self, p): tf.set_random_seed(8372749040) stt_enc = encoder.AsrEncoder(p) batch = py_utils.NestedMap() batch.src_inputs = tf.random_normal([2, 20, 16, 3], seed=92837472) batch.paddings = tf.zeros([2, 20]) return stt_enc.FPropDefaultTheta(batch)
def testEncoderConstruction(self): vn_config = py_utils.VariationalNoiseParams(None, True, False, seed=12345) p = self._EncoderParams(vn_config) _ = encoder.AsrEncoder(p)