f7c759ca562303127a9991574d5a985d4dff99e8,sonnet/python/modules/gated_rnn_test.py,LSTMTest,testSameInStaticAndDynamicWithBatchNorm,#LSTMTest#,541

Before Change

    test_static_output = tf.stack(test_static_output_unpacked, axis=1)

    with self.test_session() as session:
      tf.global_variables_initializer().run()

      def check_static_and_dynamic(training):
        // Check that static and dynamic give the same output

After Change

      return static_output, dynamic_output

    ops = connect(training=True)
    self.evaluate(tf.global_variables_initializer())
    static_out, dynamic_out = self.evaluate(ops)
    self.assertAllClose(static_out, dynamic_out)

Instances