============================= test session starts ==============================
platform linux -- Python 3.7.7, pytest-5.4.3, py-1.9.0, pluggy-0.13.1
rootdir: /home/aryaman4/sonnet
collected 3017 items

docs/ext/link_tf_api_test.py .....                                       [  0%]
examples/simple_mnist_test.py s..                                        [  0%]
sonnet/src/axis_norm_test.py ..............................s             [  1%]
sonnet/src/base_test.py ...s..................s...s........s...s...sss.. [  2%]
.......s.s                                                               [  3%]
sonnet/src/batch_apply_test.py ..s.........s................s........    [  4%]
sonnet/src/batch_norm_test.py .......................s.s                 [  5%]
sonnet/src/bias_test.py ..........s                                      [  5%]
sonnet/src/conv_test.py ..............Fs....s....s....s....s             [  6%]
sonnet/src/conv_transpose_test.py .............s....s....ss....s....s    [  8%]
sonnet/src/custom_getter_test.py .s.s                                    [  8%]
sonnet/src/deferred_test.py ........s....                                [  8%]
sonnet/src/depthwise_conv_test.py ...............s....s                  [  9%]
sonnet/src/dropout_test.py ..........s.............                      [ 10%]
sonnet/src/embed_test.py .............s.....                             [ 10%]
sonnet/src/group_norm_test.py ..............................s            [ 11%]
sonnet/src/initializers_test.py s................s......s......s......s. [ 13%]
.....................s.............s.................................... [ 15%]
........s                                                                [ 15%]
sonnet/src/leaky_clip_by_value_test.py ......s                           [ 16%]
sonnet/src/linear_test.py ...................s                           [ 16%]
sonnet/src/metrics_test.py ...s...s                                      [ 16%]
sonnet/src/mixed_precision_test.py ..............................s...... [ 18%]
.s                                                                       [ 18%]
sonnet/src/moving_averages_test.py ...........s                          [ 18%]
sonnet/src/once_test.py .............                                    [ 19%]
sonnet/src/pad_test.py ....................s........                     [ 20%]
sonnet/src/parallel_linear_test.py ..s                                   [ 20%]
sonnet/src/recurrent_test.py ....s........s..............s........s..... [ 21%]
...s....ssssss.....s........................................s.s.s        [ 23%]
sonnet/src/regularizers_test.py ...s...s...s                             [ 24%]
sonnet/src/reshape_test.py ...............................s............. [ 25%]
....s                                                                    [ 25%]
sonnet/src/scale_gradient_test.py ............s                          [ 26%]
sonnet/src/sequential_test.py ...................s.                      [ 26%]
sonnet/src/test_utils.py s                                               [ 26%]
sonnet/src/utils_test.py ............s...s...............s..........s... [ 28%]
.s..................s..s...s                                             [ 29%]
sonnet/src/conformance/api_test.py s..                                   [ 29%]
sonnet/src/conformance/build_test.py ................................... [ 30%]
.s                                                                       [ 30%]
sonnet/src/conformance/checkpoint_test.py .............................. [ 31%]
........................................................................ [ 34%]
..........................................................s.s.s.s.s.s.s. [ 36%]
s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s..ss. [ 38%]
.ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss. [ 41%]
.ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss. [ 43%]
.ss..ss..ss.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s..ss..ss..ss..s [ 46%]
s..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..s [ 48%]
s..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..s [ 50%]
s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s..ss..ss..ss..ss..ss..ss..ss..ss..ss. [ 53%]
.ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss. [ 55%]
.ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..ss..sss                     [ 57%]
sonnet/src/conformance/copy_test.py .................................... [ 58%]
....s                                                                    [ 58%]
sonnet/src/conformance/descriptors_test.py .s                            [ 58%]
sonnet/src/conformance/distribute_test.py s.s.s.s.s.s.s.s.s.s.s.s.s.s.s. [ 59%]
s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s.s. [ 62%]
s.s                                                                      [ 62%]
sonnet/src/conformance/doctest_test.py ..s.ssss.s                        [ 62%]
sonnet/src/conformance/function_test.py ................................ [ 63%]
........................................................s............... [ 65%]
........................................................................ [ 68%]
........................................................................ [ 70%]
...............................                                          [ 71%]
sonnet/src/conformance/goldens_test.py .s                                [ 71%]
sonnet/src/conformance/keras_test.py ................................... [ 73%]
......s................................................................. [ 75%]
........                                                                 [ 75%]
sonnet/src/conformance/optimizer_test.py s....................ss.sss.... [ 76%]
..............................ss.......s.s                               [ 78%]
sonnet/src/conformance/pickle_test.py .................................. [ 79%]
......s                                                                  [ 79%]
sonnet/src/conformance/saved_model_test.py ............................. [ 80%]
...........s                                                             [ 80%]
sonnet/src/conformance/tensorflow1_test.py ss                            [ 80%]
sonnet/src/conformance/xla_test.py ..................................... [ 82%]
...........................................s                             [ 83%]
sonnet/src/distribute/replicator_test.py ......ssssss......ss..s..ss..ss [ 84%]
..ss..ss                                                                 [ 84%]
sonnet/src/nets/cifar10_convnet_test.py ..........s                      [ 85%]
sonnet/src/nets/mlp_test.py ..................................s......... [ 86%]
                                                                         [ 86%]
sonnet/src/nets/resnet_test.py ....s....                                 [ 86%]
sonnet/src/nets/vqvae_test.py ..........s                                [ 87%]
sonnet/src/nets/dnc/control_test.py ..s..s....s                          [ 87%]
sonnet/src/nets/dnc/read_test.py ..s                                     [ 87%]
sonnet/src/nets/dnc/util_test.py ............s.s.s.......s               [ 88%]
sonnet/src/nets/dnc/write_test.py ..s..s..s..s                           [ 89%]
sonnet/src/optimizers/adam_test.py ..................................... [ 90%]
............................................s...........s.sss.ss         [ 92%]
sonnet/src/optimizers/momentum_test.py ................................. [ 93%]
.....................s.............s.s..ss.s...sss                       [ 95%]
sonnet/src/optimizers/rmsprop_test.py .................................. [ 96%]
..............................................................s......... [ 98%]
....s.s..ssss...s.s                                                      [ 99%]
sonnet/src/optimizers/sgd_test.py ..........s..sss...s.s                 [100%]

=================================== FAILURES ===================================
_____________________ ConvTest.testPaddingFunctionReached ______________________
../anaconda3/envs/sonnet/lib/python3.7/site-packages/mock/mock.py:1368: in patched
    keywargs) as (newargs, newkeywargs):
../anaconda3/envs/sonnet/lib/python3.7/contextlib.py:112: in __enter__
    return next(self.gen)
../anaconda3/envs/sonnet/lib/python3.7/site-packages/mock/mock.py:1334: in decoration_helper
    arg = patching.__enter__()
../anaconda3/envs/sonnet/lib/python3.7/site-packages/mock/mock.py:1437: in __enter__
    original, local = self.get_original()
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <mock.mock._patch object at 0x7f698fc90110>

    def get_original(self):
        target = self.getter()
        name = self.attribute
    
        original = DEFAULT
        local = False
    
        try:
            original = target.__dict__[name]
        except (AttributeError, KeyError):
            original = getattr(target, name, DEFAULT)
        else:
            local = True
    
        if name in _builtins and isinstance(target, ModuleType):
            self.create = True
    
        if not self.create and original is DEFAULT:
            raise AttributeError(
>               "%s does not have the attribute %r" % (target, name)
            )
E           AttributeError: <module '__main__' from '/home/aryaman4/anaconda3/envs/sonnet/bin/pytest'> does not have the attribute 'foo'

../anaconda3/envs/sonnet/lib/python3.7/site-packages/mock/mock.py:1411: AttributeError
=============================== warnings summary ===============================
/home/aryaman4/anaconda3/envs/sonnet/lib/python3.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py:15
  /home/aryaman4/anaconda3/envs/sonnet/lib/python3.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py:15: DeprecationWarning: the imp module is deprecated in favour of importlib; see the module's documentation for alternative uses
    import imp

sonnet/src/batch_norm_test.py:265
  /home/aryaman4/sonnet/sonnet/src/batch_norm_test.py:265: PytestCollectionWarning: cannot collect test class 'TestMetric' because it has a __init__ constructor (from: sonnet/src/batch_norm_test.py)
    class TestMetric(object):

sonnet/src/conformance/checkpoint_test.py:34
  /home/aryaman4/sonnet/sonnet/src/conformance/checkpoint_test.py:34: PytestCollectionWarning: cannot collect test class 'TestCheckpoint' because it has a __init__ constructor (from: sonnet/src/conformance/checkpoint_test.py)
    class TestCheckpoint(object):

examples/simple_mnist_test.py: 2 tests with warnings
sonnet/src/axis_norm_test.py: 4 tests with warnings
sonnet/src/batch_norm_test.py: 48 tests with warnings
sonnet/src/conv_test.py: 10 tests with warnings
sonnet/src/conv_transpose_test.py: 18 tests with warnings
sonnet/src/depthwise_conv_test.py: 30 tests with warnings
sonnet/src/dropout_test.py: 4 tests with warnings
sonnet/src/group_norm_test.py: 8 tests with warnings
sonnet/src/initializers_test.py: 19 tests with warnings
sonnet/src/linear_test.py: 2 tests with warnings
sonnet/src/recurrent_test.py: 533 tests with warnings
sonnet/src/reshape_test.py: 13 tests with warnings
sonnet/src/conformance/build_test.py: 577 tests with warnings
sonnet/src/conformance/checkpoint_test.py: 931 tests with warnings
sonnet/src/conformance/distribute_test.py: 434 tests with warnings
sonnet/src/conformance/doctest_test.py: 2 tests with warnings
sonnet/src/conformance/function_test.py: 3820 tests with warnings
sonnet/src/conformance/keras_test.py: 1795 tests with warnings
sonnet/src/conformance/optimizer_test.py: 1258 tests with warnings
sonnet/src/conformance/saved_model_test.py: 130 tests with warnings
sonnet/src/conformance/xla_test.py: 334 tests with warnings
sonnet/src/nets/cifar10_convnet_test.py: 334 tests with warnings
sonnet/src/nets/resnet_test.py: 312 tests with warnings
sonnet/src/nets/vqvae_test.py: 18 tests with warnings
  /home/aryaman4/anaconda3/envs/sonnet/lib/python3.7/site-packages/tensorflow/python/framework/tensor_util.py:523: DeprecationWarning: tostring() is deprecated. Use tobytes() instead.
    tensor_proto.tensor_content = nparray.tostring()

examples/simple_mnist_test.py::SimpleMnistTest::test_test_accuracy
examples/simple_mnist_test.py::SimpleMnistTest::test_train_epoch
  /home/aryaman4/anaconda3/envs/sonnet/lib/python3.7/site-packages/tensorflow/python/autograph/operators/control_flow.py:673: UserWarning: Seed 87654321 from outer graph might be getting used by function tf_data_experimental_scan_scan_body, if the random op has not been provided any seed. Explicitly set the seed in the function if this is not the intended behavior.
    ds = _general_purpose_scan(ds, init_vars, scan_body)

sonnet/src/axis_norm_test.py::LayerNormTest::test3DDataFormatAgnostic
  /home/aryaman4/sonnet/sonnet/src/axis_norm.py:110: DeprecationWarning: Using or importing the ABCs from 'collections' instead of from 'collections.abc' is deprecated since Python 3.3,and in 3.9 it will stop working
    elif (isinstance(axis, collections.Iterable) and

sonnet/src/batch_apply_test.py::BatchApplyTest::test_kwargs
sonnet/src/batch_apply_test.py::BatchApplyTest::test_kwargs
sonnet/src/batch_apply_test.py::BatchApplyTest::test_kwargs
  /home/aryaman4/anaconda3/envs/sonnet/lib/python3.7/site-packages/tree/__init__.py:312: DeprecationWarning: Using or importing the ABCs from 'collections' instead of from 'collections.abc' is deprecated since Python 3.3,and in 3.9 it will stop working
    return _tree.flatten(structure)

sonnet/src/conv_transpose_test.py::ConvTransposeTest::testGivenOutputShape
  /home/aryaman4/anaconda3/envs/sonnet/lib/python3.7/site-packages/tensorflow/python/ops/nn_ops.py:2704: DeprecationWarning: Using or importing the ABCs from 'collections' instead of from 'collections.abc' is deprecated since Python 3.3,and in 3.9 it will stop working
    elif isinstance(output_shape, collections.Sized):

sonnet/src/once_test.py::OnceTest::test_always_returns_none
  /home/aryaman4/sonnet/sonnet/src/once_test.py:44: DeprecationWarning: Please use assertRaisesRegex instead.
    with self.assertRaisesRegexp(ValueError, "snt.once .* cannot return"):

sonnet/src/recurrent_test.py: 18 tests with warnings
  /home/aryaman4/anaconda3/envs/sonnet/lib/python3.7/site-packages/tensorflow/python/framework/test_util.py:2352: VisibleDeprecationWarning: Creating an ndarray from ragged nested sequences (which is a list-or-tuple of lists-or-tuples-or ndarrays with different lengths or shapes) is deprecated. If you meant to do this, you must specify 'dtype=object' when creating the ndarray
    return np.array(a)

-- Docs: https://docs.pytest.org/en/latest/warnings.html
============================ slowest test durations ============================
4.44s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_resnet50_Replicator_use_function
4.35s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_resnet50_Replicator_use_function
4.30s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_resnet50_Replicator
3.91s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_nets.ResNet50
3.58s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_resnet50
3.54s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.ResNet50_construct_module_in_function
3.38s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_nets.ResNet50
3.38s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(LSTM, dynamic_unroll)_autograph
3.36s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.ResNet50_autograph
3.36s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_resnet50
3.35s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_nets.ResNet50
3.27s call     sonnet/src/nets/resnet_test.py::ResnetTest::test_tf_function1
3.23s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_resnet50
3.23s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.ResNet50_autograph
3.16s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_nets.ResNet50
3.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.ResNet50_autograph
2.97s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.ResNet50_not_autograph
2.92s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.ResNet50_not_autograph
2.90s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.ResNet50_not_construct_module_in_function
2.90s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.ResNet50_not_autograph
2.63s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_resnet50_Replicator_not_use_function
2.62s call     sonnet/src/nets/resnet_test.py::ResnetTest::test_tf_function0
2.60s call     sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testDefunBackProp
2.51s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.Cifar10ConvNet_autograph
2.50s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_nets.Cifar10ConvNet
2.40s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.Cifar10ConvNet_construct_module_in_function
2.38s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator
2.28s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.Cifar10ConvNet_autograph
2.27s call     sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testModelForwardsFunction1
2.17s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_nets.Cifar10ConvNet
2.16s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.Cifar10ConvNet_not_autograph
2.16s call     examples/simple_mnist_test.py::SimpleMnistTest::test_test_accuracy
2.13s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_nets.Cifar10ConvNet
2.13s call     sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testModelForwardsFunction2
2.12s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_lstm_8_projected_1_Replicator_use_function
2.10s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_with_updating_module
2.10s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.Cifar10ConvNet_not_autograph
2.09s call     sonnet/src/recurrent_test.py::UnrolledLSTMTest::testComputationAgainstLSTM0
2.08s call     sonnet/src/conformance/doctest_test.py::DoctestTest::test_doctest_sonnet.distribute
2.07s call     sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testModelForwardsFunction0
2.07s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.Cifar10ConvNet_autograph
2.07s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.Cifar10ConvNet_not_construct_module_in_function
2.05s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_sum_2x2_Replicator
2.05s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_nets.Cifar10ConvNet
2.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.Cifar10ConvNet_not_autograph
1.80s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_nets.ResNet50_autograph
1.78s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_unrolled_lstm_1_Replicator
1.58s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_nets.ResNet50_not_autograph
1.56s call     sonnet/src/recurrent_test.py::UnrolledLSTMTest::testNumStepsPolymorphism0
1.50s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_resnet50_Replicator_use_function
1.37s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(UnrolledLSTM)
1.21s call     sonnet/src/batch_norm_test.py::BaseBatchNormTest::testWithTfFunction0
1.17s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_nets.Cifar10ConvNet_autograph
1.17s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_gru_1_Replicator
1.13s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_ema_train_Replicator_use_function
1.12s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv2d_lstm_3x3_2x2_Replicator
1.11s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_lstm_1_Replicator_use_function
1.10s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_vanilla_rnn_8_Replicator
1.10s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_dropout_Replicator_use_function
1.09s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_group_norm_2_1x3x4_Replicator
1.08s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_ema_2_Replicator_use_function
1.08s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_ema_eval_Replicator_use_function
1.07s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_depthwise_conv2d_3x3_2x2_Replicator
1.07s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_linear_1x1_Replicator
1.07s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_base_batch_norm_scale_offset_1x2x2x3_Replicator
1.06s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_nets.Cifar10ConvNet_not_autograph
1.05s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_linear_nobias_1x1_Replicator
1.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mlp_nobias_3x4x5_1x3_Replicator_use_function
1.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_group_norm_2_1x3x4_Replicator_use_function
1.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_sum_2x2_Replicator_use_function
1.02s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_trainable_state_Replicator
1.00s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(UnrolledLSTM)_autograph
0.99s call     sonnet/src/conv_transpose_test.py::ConvTransposeTest::testGraphConv0
0.96s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(UnrolledLSTM)_not_autograph
0.94s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(UnrolledLSTM)
0.88s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_resnet50_Replicator
0.88s call     sonnet/src/recurrent_test.py::UnrollTest::testEmptyOutputs2
0.83s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.Adam_autograph
0.82s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_unrolled_lstm_1_Replicator_use_function
0.81s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_unrolled_lstm_1_Replicator_use_function
0.80s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(UnrolledLSTM)_autograph
0.80s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(UnrolledLSTM)_not_construct_module_in_function
0.80s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(UnrolledLSTM)_construct_module_in_function
0.79s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(UnrolledLSTM)_autograph
0.78s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Recurrent(UnrolledLSTM)
0.77s call     sonnet/src/recurrent_test.py::UnrolledLSTMTest::testComputationAgainstLSTM2
0.76s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(UnrolledLSTM)_not_autograph
0.76s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_unrolled_lstm_1
0.76s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(UnrolledLSTM)_not_autograph
0.75s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_resnet50_Replicator_not_use_function
0.74s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_resnet50
0.74s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(UnrolledLSTM)
0.73s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.RMSProp_autograph
0.73s call     examples/simple_mnist_test.py::SimpleMnistTest::test_train_epoch
0.71s call     sonnet/src/conv_test.py::ConvTest::testFunctionSamePaddingNoBias
0.69s call     sonnet/src/recurrent_test.py::DeepRNNTest::testWithSkipConnectionsOutputs1
0.67s call     sonnet/src/group_norm_test.py::GroupNormTest::testBatchSizeAgnostic
0.66s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_resnet50
0.59s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_resnet50_Replicator_not_use_function
0.59s call     sonnet/src/initializers_test.py::OrthogonalInitializerTest::testBatchAgnostic
0.56s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.0,epsilon=1e-07,centered=True,seed=28
0.56s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.0,epsilon=1e-07,centered=True,seed=90
0.56s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.0,epsilon=1e-08,centered=True,seed=2
0.56s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.0,epsilon=1e-07,centered=True,seed=2
0.56s call     sonnet/src/axis_norm_test.py::LayerNormTest::testShapeAgnostic
0.56s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.0,epsilon=1e-08,centered=True,seed=90
0.56s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.0,epsilon=1e-08,centered=True,seed=2
0.56s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.0,epsilon=1e-07,centered=True,seed=28
0.56s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.0,epsilon=1e-08,centered=True,seed=28
0.56s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.0,epsilon=1e-07,centered=True,seed=90
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.0,epsilon=1e-08,centered=True,seed=2
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.0,epsilon=1e-08,centered=True,seed=90
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.0,epsilon=1e-08,centered=True,seed=90
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.0,epsilon=1e-07,centered=True,seed=28
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.0,epsilon=1e-07,centered=True,seed=2
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.0,epsilon=1e-08,centered=True,seed=2
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.0,epsilon=1e-08,centered=True,seed=28
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.0,epsilon=1e-07,centered=True,seed=2
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.0,epsilon=1e-08,centered=True,seed=90
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.0,epsilon=1e-08,centered=True,seed=28
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.0,epsilon=1e-07,centered=True,seed=90
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.0,epsilon=1e-07,centered=True,seed=28
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.0,epsilon=1e-08,centered=True,seed=28
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.0,epsilon=1e-07,centered=True,seed=2
0.55s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.0,epsilon=1e-07,centered=True,seed=90
0.55s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(Conv1DLSTM, dynamic_unroll)_autograph
0.53s call     sonnet/src/recurrent_test.py::LSTMTest::testComputationAgainstNumPy4
0.52s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Bias
0.52s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_resnet50
0.50s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_unrolled_lstm_1
0.50s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv3DLSTM, dynamic_unroll)_construct_module_in_function
0.49s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_resnet50
0.48s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic0
0.48s call     sonnet/src/recurrent_test.py::DeepRNNTest::testComputationAgainstNumPy1
0.47s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_BatchNorm_autograph
0.47s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv1DLSTM, dynamic_unroll)_autograph
0.46s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Recurrent(Conv1DLSTM, dynamic_unroll)
0.45s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.Momentum_autograph
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.0,epsilon=1e-07,centered=False,seed=2
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.0,epsilon=1e-07,centered=False,seed=28
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.0,epsilon=1e-08,centered=False,seed=2
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.0,epsilon=1e-07,centered=False,seed=2
0.44s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_nets.MLP
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.0,epsilon=1e-08,centered=False,seed=28
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.0,epsilon=1e-07,centered=False,seed=2
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.0,epsilon=1e-07,centered=False,seed=2
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.99,beta_2=0.99,epsilon=1e-08,seed=2
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.0,epsilon=1e-07,centered=False,seed=90
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.0,epsilon=1e-07,centered=False,seed=28
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.99,beta_2=0.99,epsilon=1e-08,seed=2
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.999,beta_2=0.9,epsilon=1e-08,seed=90
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.0,epsilon=1e-07,centered=False,seed=90
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.0,epsilon=1e-08,centered=False,seed=90
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.0,epsilon=1e-08,centered=False,seed=2
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.0,epsilon=1e-07,centered=False,seed=28
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.0,epsilon=1e-07,centered=False,seed=90
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.0,epsilon=1e-07,centered=False,seed=90
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.0,epsilon=1e-08,centered=False,seed=2
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.0,epsilon=1e-08,centered=False,seed=28
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.0,epsilon=1e-08,centered=False,seed=28
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.0,epsilon=1e-07,centered=False,seed=28
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.0,epsilon=1e-08,centered=False,seed=2
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.0,epsilon=1e-08,centered=False,seed=28
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.0,epsilon=1e-08,centered=False,seed=90
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.9,beta_2=0.99,epsilon=1e-08,seed=28
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.0,epsilon=1e-08,centered=False,seed=90
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.9,beta_2=0.99,epsilon=1e-08,seed=2
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.9,beta_2=0.999,epsilon=1e-08,seed=90
0.44s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.0,epsilon=1e-08,centered=False,seed=90
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.99,beta_2=0.9,epsilon=1e-08,seed=90
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.9,beta_2=0.999,epsilon=1e-08,seed=90
0.44s call     sonnet/src/conv_transpose_test.py::ConvTransposeTest::testInitializerVariance2
0.44s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(VanillaRNN, dynamic_unroll)_autograph
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.999,beta_2=0.999,epsilon=1e-08,seed=2
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.9,beta_2=0.9,epsilon=1e-08,seed=28
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.99,beta_2=0.999,epsilon=1e-08,seed=2
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.9,beta_2=0.9,epsilon=1e-08,seed=2
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.9,beta_2=0.9,epsilon=1e-08,seed=28
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.999,beta_2=0.999,epsilon=1e-08,seed=90
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.99,beta_2=0.9,epsilon=1e-08,seed=2
0.44s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.999,beta_2=0.9,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.9,beta_2=0.999,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.9,beta_2=0.999,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.999,beta_2=0.9,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.999,beta_2=0.999,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.9,beta_2=0.9,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.99,beta_2=0.99,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.9,beta_2=0.9,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.99,beta_2=0.9,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.999,beta_2=0.99,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.99,beta_2=0.99,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.999,beta_2=0.99,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.9,beta_2=0.99,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.9,beta_2=0.999,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.99,beta_2=0.999,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.99,beta_2=0.999,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.9,beta_2=0.9,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.999,beta_2=0.999,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.999,beta_2=0.999,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.9,beta_2=0.99,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.9,beta_2=0.99,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.999,beta_2=0.999,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.999,beta_2=0.9,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.99,beta_2=0.999,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.9,beta_2=0.9,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.999,beta_2=0.99,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.9,beta_2=0.9,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.9,beta_2=0.99,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.999,beta_2=0.9,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.99,beta_2=0.999,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.9,beta_2=0.99,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.9,beta_2=0.999,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.999,beta_2=0.99,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.99,beta_2=0.99,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.9,beta_2=0.9,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.99,beta_2=0.9,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.99,beta_2=0.999,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.999,beta_2=0.99,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.999,beta_2=0.9,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.999,beta_2=0.9,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.999,beta_2=0.999,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.999,beta_2=0.99,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.999,beta_2=0.99,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.99,beta_2=0.9,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.9,beta_2=0.99,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.999,beta_2=0.9,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.999,beta_2=0.99,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.99,beta_2=0.9,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.9,beta_2=0.999,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.99,beta_2=0.9,epsilon=1e-08,seed=90
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.999,beta_2=0.999,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.99,beta_2=0.999,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.999,beta_2=0.9,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.99,beta_2=0.99,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.9,beta_2=0.99,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.99,beta_2=0.999,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.999,beta_2=0.999,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.999,beta_2=0.99,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,beta_1=0.9,beta_2=0.999,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,beta_1=0.9,beta_2=0.999,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.99,beta_2=0.9,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.99,beta_2=0.9,epsilon=1e-08,seed=28
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.99,beta_2=0.99,epsilon=1e-08,seed=2
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.99,beta_2=0.999,epsilon=1e-08,seed=2
0.43s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(Conv3DLSTM, dynamic_unroll)
0.43s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.99,beta_2=0.99,epsilon=1e-08,seed=90
0.42s call     sonnet/src/optimizers/adam_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,beta_1=0.99,beta_2=0.99,epsilon=1e-08,seed=28
0.42s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_unrolled_lstm_1
0.42s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_resnet50_Replicator
0.42s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_resnet50
0.40s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(GRU, dynamic_unroll)
0.40s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_unrolled_lstm_1_Replicator_use_function
0.40s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(GRU, static_unroll)
0.40s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_nets.VectorQuantizer
0.40s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_nets.VectorQuantizerEMA
0.40s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_resnet50
0.39s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(Conv2DLSTM, dynamic_unroll)
0.39s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(Conv1DLSTM, dynamic_unroll)
0.38s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.5,epsilon=1e-08,centered=True,seed=2
0.38s call     sonnet/src/recurrent_test.py::GRUTest::testComputationAgainstNumPy1
0.37s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(Conv3DLSTM, dynamic_unroll)
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.5,epsilon=1e-08,centered=True,seed=2
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.5,epsilon=1e-07,centered=True,seed=28
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.5,epsilon=1e-08,centered=True,seed=90
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.5,epsilon=1e-07,centered=True,seed=28
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.5,epsilon=1e-07,centered=True,seed=90
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.5,epsilon=1e-07,centered=True,seed=90
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.5,epsilon=1e-08,centered=True,seed=28
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.5,epsilon=1e-08,centered=True,seed=90
0.37s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(Conv2DLSTM, dynamic_unroll)
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.5,epsilon=1e-08,centered=True,seed=2
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.5,epsilon=1e-08,centered=True,seed=2
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.5,epsilon=1e-08,centered=True,seed=28
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.5,epsilon=1e-07,centered=True,seed=2
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.5,epsilon=1e-07,centered=True,seed=90
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.5,epsilon=1e-07,centered=True,seed=2
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.5,epsilon=1e-07,centered=True,seed=2
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.5,epsilon=1e-08,centered=True,seed=28
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.5,epsilon=1e-07,centered=True,seed=28
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.5,epsilon=1e-08,centered=True,seed=90
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.5,epsilon=1e-07,centered=True,seed=28
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.5,epsilon=1e-07,centered=True,seed=2
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.5,epsilon=1e-07,centered=True,seed=90
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.5,epsilon=1e-08,centered=True,seed=28
0.37s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.5,epsilon=1e-08,centered=True,seed=90
0.34s call     sonnet/src/recurrent_test.py::UnrollTest::testEmptyOutputs3
0.34s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_CrossReplicaBatchNorm
0.34s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(GRU, dynamic_unroll)_not_construct_module_in_function
0.34s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(GRU, dynamic_unroll)_construct_module_in_function
0.33s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(LSTM, dynamic_unroll)
0.33s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Recurrent(GRU, dynamic_unroll)
0.33s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator_use_function
0.33s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(GRU, dynamic_unroll)_autograph
0.33s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.5,epsilon=1e-08,centered=False,seed=90
0.33s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(GRU, dynamic_unroll)_autograph
0.33s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_base_batch_norm_1x2x2x3_Replicator_use_function
0.33s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(GRU, dynamic_unroll)_not_autograph
0.32s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(GRU, dynamic_unroll)_not_autograph
0.32s call     sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape0
0.32s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_nets.MLP
0.32s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.5,epsilon=1e-07,centered=False,seed=90
0.32s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Linear_autograph
0.32s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.5,epsilon=1e-08,centered=False,seed=28
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.5,epsilon=1e-07,centered=False,seed=90
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.5,epsilon=1e-08,centered=False,seed=2
0.31s call     sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape_has_static_info_in_graph0
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.5,epsilon=1e-07,centered=False,seed=28
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.5,epsilon=1e-07,centered=False,seed=28
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.5,epsilon=1e-08,centered=False,seed=90
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.5,epsilon=1e-08,centered=False,seed=28
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.5,epsilon=1e-07,centered=False,seed=2
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.5,epsilon=1e-07,centered=False,seed=2
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.5,epsilon=1e-08,centered=False,seed=28
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.5,epsilon=1e-07,centered=False,seed=28
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.5,epsilon=1e-07,centered=False,seed=2
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.5,epsilon=1e-08,centered=False,seed=2
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.5,epsilon=1e-08,centered=False,seed=90
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.8,momentum=0.5,epsilon=1e-08,centered=False,seed=2
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.9,momentum=0.5,epsilon=1e-08,centered=False,seed=28
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,decay=0.8,momentum=0.5,epsilon=1e-07,centered=False,seed=90
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.5,epsilon=1e-07,centered=False,seed=90
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.5,epsilon=1e-08,centered=False,seed=2
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.5,epsilon=1e-07,centered=False,seed=28
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.5,epsilon=1e-08,centered=False,seed=90
0.31s call     sonnet/src/optimizers/rmsprop_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,decay=0.9,momentum=0.5,epsilon=1e-07,centered=False,seed=2
0.31s call     sonnet/src/recurrent_test.py::ConvNDLSTMTest::testComputationAgainstNumPy3
0.30s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv1DLSTM, dynamic_unroll)_autograph
0.30s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(VanillaRNN, dynamic_unroll)
0.30s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv1DLSTM, dynamic_unroll)_not_autograph
0.30s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_ema_train_Replicator_use_function
0.30s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv1DLSTM, dynamic_unroll)_not_construct_module_in_function
0.30s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv1DLSTM, dynamic_unroll)_construct_module_in_function
0.30s call     sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testFunctionSamePaddingNoBias
0.29s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_vqvae_ema_train
0.29s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv1DLSTM, dynamic_unroll)_not_autograph
0.29s call     sonnet/src/recurrent_test.py::UnknownStepsUnrollTest::testStaticUnroll
0.29s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(GRU, dynamic_unroll)
0.29s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv3DLSTM, dynamic_unroll)_not_construct_module_in_function
0.29s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv3DLSTM, dynamic_unroll)_autograph
0.29s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv3DLSTM, dynamic_unroll)_autograph
0.29s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_cifar10_convnet_2x3_2x2_1x3x3x2
0.29s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv2DLSTM, dynamic_unroll)_autograph
0.29s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(GRU, dynamic_unroll)
0.29s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv3DLSTM, dynamic_unroll)_not_autograph
0.28s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv2DLSTM, dynamic_unroll)_autograph
0.28s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv2DLSTM, dynamic_unroll)_not_construct_module_in_function
0.28s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(Conv1DLSTM, dynamic_unroll)
0.28s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Recurrent(LSTM, dynamic_unroll)
0.28s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv3DLSTM, dynamic_unroll)_not_autograph
0.28s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(Conv1DLSTM, dynamic_unroll)
0.28s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_vqvae_ema_train
0.28s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv2DLSTM, dynamic_unroll)_not_autograph
0.28s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv2DLSTM, dynamic_unroll)_construct_module_in_function
0.27s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.Adam_autograph
0.27s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv2DLSTM, dynamic_unroll)_not_autograph
0.27s call     sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthAllFull2
0.27s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_vqvae_ema_train
0.27s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(LSTM, dynamic_unroll)_not_autograph
0.27s call     sonnet/src/recurrent_test.py::UnknownStepsUnrollTest::testDynamicUnroll
0.27s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(Conv3DLSTM, dynamic_unroll)
0.27s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(Conv2DLSTM, dynamic_unroll)
0.27s call     sonnet/src/batch_norm_test.py::BaseBatchNormTest::testWithTfFunctionTfArgs0
0.27s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(Conv1DLSTM, dynamic_unroll)_not_autograph
0.27s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(LSTM, dynamic_unroll)_construct_module_in_function
0.27s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(LSTM, dynamic_unroll)_autograph
0.26s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(Conv3DLSTM, dynamic_unroll)_autograph
0.26s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(LSTM, dynamic_unroll)_autograph
0.26s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(LSTM, dynamic_unroll)_not_construct_module_in_function
0.26s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(Conv2DLSTM, dynamic_unroll)_autograph
0.26s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(GRU, dynamic_unroll)_autograph
0.26s call     sonnet/src/batch_norm_test.py::BaseBatchNormTest::testWithTfFunctionTfArgs1
0.26s call     sonnet/src/batch_norm_test.py::BaseBatchNormTest::testWithTfFunction1
0.26s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(LSTM, dynamic_unroll)_not_autograph
0.26s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(LSTM, dynamic_unroll)_not_autograph
0.26s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator_use_function
0.25s call     sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testResetWithTFFunction0
0.25s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(GRU, dynamic_unroll)_not_autograph
0.25s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Recurrent(Conv3DLSTM, dynamic_unroll)
0.25s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(Conv3DLSTM, dynamic_unroll)_not_autograph
0.25s call     sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthAllEmpty2
0.25s call     sonnet/src/recurrent_test.py::VanillaRNNTest::testComputationAgainstNumPy1
0.25s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Recurrent(Conv2DLSTM, dynamic_unroll)
0.25s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(Conv2DLSTM, dynamic_unroll)_not_autograph
0.25s call     sonnet/src/recurrent_test.py::UnrollTest::testNestedOutputs2
0.24s call     sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthAllFull3
0.24s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_BatchNorm
0.24s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Recurrent(VanillaRNN, dynamic_unroll)
0.24s call     sonnet/src/recurrent_test.py::UnrollTest::testFlat2
0.24s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(LSTM, dynamic_unroll)
0.24s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(LSTM, dynamic_unroll)
0.24s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Recurrent(LSTM, static_unroll)
0.24s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(VanillaRNN, static_unroll)_autograph
0.24s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.9,use_nesterov=True,seed=28
0.24s call     sonnet/src/recurrent_test.py::UnrollTest::testNestedInputs2
0.24s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.5,use_nesterov=True,seed=90
0.24s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.2,use_nesterov=True,seed=90
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.9,use_nesterov=True,seed=90
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.2,use_nesterov=True,seed=28
0.23s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.VectorQuantizerEMA_construct_module_in_function
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.2,use_nesterov=True,seed=28
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.2,use_nesterov=True,seed=90
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.5,use_nesterov=True,seed=2
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.5,use_nesterov=True,seed=28
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.5,use_nesterov=True,seed=2
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.5,use_nesterov=True,seed=90
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.2,use_nesterov=True,seed=2
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.5,use_nesterov=True,seed=90
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.5,use_nesterov=True,seed=2
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.5,use_nesterov=True,seed=28
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.5,use_nesterov=True,seed=28
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.9,use_nesterov=True,seed=2
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.2,use_nesterov=True,seed=90
0.23s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(VanillaRNN, dynamic_unroll)_construct_module_in_function
0.23s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(Conv1DLSTM, static_unroll)
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.9,use_nesterov=True,seed=90
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.2,use_nesterov=True,seed=2
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.2,use_nesterov=True,seed=28
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.9,use_nesterov=True,seed=28
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.9,use_nesterov=True,seed=90
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.2,use_nesterov=True,seed=2
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.9,use_nesterov=True,seed=2
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.9,use_nesterov=True,seed=2
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.9,use_nesterov=True,seed=28
0.23s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(Conv3DLSTM, static_unroll)
0.23s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(VanillaRNN, dynamic_unroll)_autograph
0.23s call     sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape_has_static_info_in_graph0
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.2,use_nesterov=False,seed=2
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.2,use_nesterov=False,seed=2
0.23s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(VanillaRNN, dynamic_unroll)_not_construct_module_in_function
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.5,use_nesterov=False,seed=90
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.2,use_nesterov=False,seed=90
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.2,use_nesterov=False,seed=28
0.23s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(VanillaRNN, dynamic_unroll)_not_autograph
0.23s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv3d_lstm_3x3_2x2
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.2,use_nesterov=False,seed=90
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.5,use_nesterov=False,seed=2
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.2,use_nesterov=False,seed=28
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.5,use_nesterov=False,seed=28
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.2,use_nesterov=False,seed=28
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.5,use_nesterov=False,seed=90
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.9,use_nesterov=False,seed=2
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.9,use_nesterov=False,seed=28
0.23s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.9,use_nesterov=False,seed=28
0.22s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.9,use_nesterov=False,seed=90
0.22s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.5,use_nesterov=False,seed=90
0.22s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.5,use_nesterov=False,seed=2
0.22s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.9,use_nesterov=False,seed=2
0.22s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.9,use_nesterov=False,seed=90
0.22s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.9,use_nesterov=False,seed=2
0.22s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.9,use_nesterov=False,seed=90
0.22s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.001,momentum=0.9,use_nesterov=False,seed=28
0.22s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.5,use_nesterov=False,seed=28
0.22s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.2,use_nesterov=False,seed=90
0.22s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(VanillaRNN, dynamic_unroll)_not_autograph
0.22s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.1,momentum=0.5,use_nesterov=False,seed=28
0.22s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.5,use_nesterov=False,seed=2
0.22s call     sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlowlearning_rate=0.01,momentum=0.2,use_nesterov=False,seed=2
0.22s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(Conv2DLSTM, static_unroll)
0.22s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(VanillaRNN, dynamic_unroll)
0.22s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv3d_lstm_3x3_2x2
0.22s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(VanillaRNN, dynamic_unroll)
0.22s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_ema_eval_Replicator_use_function
0.22s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_training_1x2x2x3_Replicator_use_function
0.21s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_CrossReplicaBatchNorm_autograph
0.21s call     sonnet/src/initializers_test.py::OrthogonalInitializerTest::testTFFunctionSimple
0.21s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_Replicator_use_function
0.21s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_vqvae
0.21s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_CrossReplicaBatchNorm
0.21s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_vqvae
0.21s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_base_batch_norm_1x2x2x3_Replicator
0.21s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv3d_transpose_3x3_2x2
0.20s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv3d_transpose_3x3_2x2
0.20s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_vqvae_ema_eval
0.20s call     sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testModelForwards2
0.20s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_cifar10_convnet_2x3_2x2_1x3x3x2
0.20s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_batch_norm_training_1x2x2x3
0.20s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(VanillaRNN, dynamic_unroll)_autograph
0.19s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(LSTM, static_unroll)
0.19s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_CrossReplicaBatchNorm_not_autograph
0.19s call     sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testDifferentSizedImages
0.19s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_base_batch_norm_1x2x2x3
0.19s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.MLP_autograph
0.19s call     sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthRange2
0.19s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_lstm_3x3_2x2_Replicator_use_function
0.19s call     sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthOneZeroLength3
0.19s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(GRU, static_unroll)_autograph
0.19s call     sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthAllEmpty3
0.18s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(GRU, static_unroll)
0.18s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_BatchNorm
0.18s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv1d_3x3_2x2_Replicator
0.18s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(GRU, static_unroll)_autograph
0.18s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_batch_norm_training_1x2x2x3_Replicator
0.18s call     sonnet/src/conformance/doctest_test.py::DoctestTest::test_doctest_sonnet
0.18s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(GRU, static_unroll)
0.18s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(VanillaRNN, dynamic_unroll)_not_autograph
0.18s call     sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape0
0.18s call     sonnet/src/initializers_test.py::IdentityInitializerTest::testBatchAgnostic
0.18s call     sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthRange3
0.18s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_BatchNorm_not_autograph
0.18s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(Conv3DLSTM, static_unroll)
0.18s call     sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthOneZeroLength2
0.18s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(Conv1DLSTM, static_unroll)
0.17s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(Conv3DLSTM, static_unroll)
0.17s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_CrossReplicaBatchNorm_construct_module_in_function
0.17s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(Conv1DLSTM, static_unroll)
0.17s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_base_batch_norm_1x2x2x3_Replicator_use_function
0.17s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_training_1x2x2x3_Replicator_use_function
0.17s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Recurrent(GRU, static_unroll)
0.17s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv2d_lstm_3x3_2x2
0.17s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(VanillaRNN, static_unroll)
0.17s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv2d_lstm_3x3_2x2
0.17s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(Conv2DLSTM, static_unroll)
0.17s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv1DLSTM, static_unroll)_autograph
0.17s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(Conv2DLSTM, static_unroll)
0.17s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv3DLSTM, static_unroll)_autograph
0.17s call     sonnet/src/conformance/keras_test.py::KerasTest::test_layer_with_model
0.17s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv1DLSTM, static_unroll)_autograph
0.17s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_CrossReplicaBatchNorm
0.17s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_CrossReplicaBatchNorm
0.17s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_vqvae
0.17s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(LSTM, static_unroll)_autograph
0.17s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv3DLSTM, static_unroll)_autograph
0.17s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(GRU, static_unroll)_construct_module_in_function
0.16s call     sonnet/src/nets/vqvae_test.py::VqvaeTest::testEmaUpdating0
0.16s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_nets.MLP
0.16s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(GRU, static_unroll)_not_construct_module_in_function
0.16s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv2DLSTM, static_unroll)_autograph
0.16s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv2DLSTM, static_unroll)_autograph
0.16s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_CrossReplicaBatchNorm_autograph
0.16s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_CrossReplicaBatchNorm_autograph
0.16s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_BatchNorm_construct_module_in_function
0.16s call     sonnet/src/nets/vqvae_test.py::VqvaeTest::testEmaUpdating1
0.16s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_CrossReplicaBatchNorm_not_construct_module_in_function
0.16s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_CrossReplicaBatchNorm_not_autograph
0.16s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_CrossReplicaBatchNorm_not_autograph
0.16s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Recurrent(Conv1DLSTM, static_unroll)
0.16s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_lstm_3x3_2x2_Replicator_use_function
0.16s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_gru_1
0.16s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Recurrent(Conv3DLSTM, static_unroll)
0.16s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(GRU, static_unroll)_not_autograph
0.16s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(GRU, static_unroll)_not_autograph
0.15s call     sonnet/src/recurrent_test.py::UnrollTest::testNestedOutputs3
0.15s call     sonnet/src/nets/resnet_test.py::ResnetTest::test_simple1
0.15s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_mlp_3x4x5_1x3_Replicator
0.15s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_nets.VectorQuantizerEMA
0.15s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Recurrent(Conv2DLSTM, static_unroll)
0.15s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(LSTM, static_unroll)
0.15s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_BatchNorm
0.15s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_scale_offset_1x2x2x3_Replicator_use_function
0.15s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_BatchNorm
0.15s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_nets.MLP
0.15s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.MLP_autograph
0.15s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(LSTM, static_unroll)_autograph
0.15s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(LSTM, static_unroll)
0.15s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(LSTM, static_unroll)_autograph
0.15s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_ema_2_Replicator_use_function
0.15s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.VectorQuantizerEMA_autograph
0.14s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_BatchNorm_autograph
0.14s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_BatchNorm_autograph
0.14s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_gru_1_Replicator_use_function
0.14s call     sonnet/src/recurrent_test.py::UnrollTest::testNestedInputs3
0.14s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_lstm_8_projected_1_Replicator
0.14s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_BatchNorm_not_construct_module_in_function
0.14s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(Conv1DLSTM, static_unroll)_autograph
0.14s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_ema_train_Replicator_use_function
0.14s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv1DTranspose_autograph
0.14s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_mlp_3x4x5_1x3
0.14s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_trainable_state_Replicator_use_function
0.14s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_base_batch_norm_1x2x2x3_Replicator_use_function
0.14s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(Conv3DLSTM, static_unroll)_autograph
0.14s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_BatchNorm_not_autograph
0.14s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mlp_3x4x5_1x3_Replicator_use_function
0.14s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_BatchNorm_not_autograph
0.14s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv1d_lstm_3x3_2x2
0.14s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_BatchNorm_autograph
0.14s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_vqvae_ema_eval
0.14s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv3DLSTM, static_unroll)_construct_module_in_function
0.14s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv3DLSTM, static_unroll)_not_construct_module_in_function
0.14s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(Conv2DLSTM, static_unroll)_autograph
0.14s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.MLP_not_autograph
0.14s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv1DLSTM, static_unroll)_not_construct_module_in_function
0.14s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_cifar10_convnet_2x3_2x2_1x3x3x2
0.14s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.MLP_autograph
0.13s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv3DLSTM, static_unroll)_not_autograph
0.13s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv1DLSTM, static_unroll)_construct_module_in_function
0.13s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_batch_norm_scale_offset_1x2x2x3
0.13s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_vqvae_ema_eval
0.13s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(VanillaRNN, static_unroll)
0.13s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv2DLSTM, static_unroll)_construct_module_in_function
0.13s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(VanillaRNN, static_unroll)
0.13s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_batch_norm_1x2x2x3
0.13s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv2d_lstm_3x3_2x2
0.13s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv1DLSTM, static_unroll)_not_autograph
0.13s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv1d_lstm_3x3_2x2_Replicator
0.13s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_gru_1_Replicator_use_function
0.13s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv2DLSTM, static_unroll)_not_construct_module_in_function
0.13s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv3d_lstm_3x3_2x2
0.13s call     sonnet/src/nets/resnet_test.py::ResnetTest::test_simple0
0.13s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv1DLSTM, static_unroll)_not_autograph
0.13s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(VanillaRNN, static_unroll)_autograph
0.13s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv3DLSTM, static_unroll)_not_autograph
0.13s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_1x2x2x3_Replicator_use_function
0.13s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_batch_norm_scale_offset_1x2x2x3_Replicator
0.13s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_lstm_3x3_2x2_Replicator_use_function
0.13s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_lstm_8_projected_1
0.13s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_gru_1
0.13s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(LSTM, static_unroll)_construct_module_in_function
0.13s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_batch_norm_1x2x2x3_Replicator
0.13s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv2DLSTM, static_unroll)_not_autograph
0.13s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_base_batch_norm_1x2x2x3
0.13s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(LSTM, static_unroll)_not_construct_module_in_function
0.12s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_gru_1
0.12s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv3DTranspose_autograph
0.12s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv2DLSTM, static_unroll)_not_autograph
0.12s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv2DTranspose_autograph
0.12s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(VanillaRNN, static_unroll)_autograph
0.12s call     sonnet/src/recurrent_test.py::LSTMTest::testComputationAgainstNumPy7
0.12s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Conv1DTranspose
0.12s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_lstm_1
0.12s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(LSTM, static_unroll)_not_autograph
0.12s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv3d_lstm_3x3_2x2_Replicator
0.12s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(LSTM, static_unroll)_not_autograph
0.12s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(LSTM, static_unroll)_not_autograph
0.12s call     sonnet/src/recurrent_test.py::LSTMTest::testComputationAgainstNumPy6
0.12s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.MLP_construct_module_in_function
0.12s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_dropout
0.12s call     sonnet/src/conformance/keras_test.py::KerasTest::test_layer_adapter_custom_method
0.12s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_1x2x2x3_Replicator_use_function
0.12s call     sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_eager_variable_creator0
0.12s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_scale_offset_1x2x2x3_Replicator_use_function
0.12s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(GRU, static_unroll)_autograph
0.12s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator_use_function
0.12s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_lstm_3x3_2x2_Replicator_use_function
0.12s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_lstm_3x3_2x2_Replicator_use_function
0.12s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_ema_2
0.12s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_lstm_1_Replicator
0.12s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.MLP_not_construct_module_in_function
0.12s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Recurrent(VanillaRNN, static_unroll)
0.11s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv1d_transpose_3x3_2x2
0.11s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv1d_lstm_3x3_2x2
0.11s call     sonnet/src/recurrent_test.py::UnrollTest::testFlat3
0.11s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_lstm_3x3_2x2_Replicator_use_function
0.11s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.VectorQuantizer_construct_module_in_function
0.11s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.VectorQuantizerEMA_not_autograph
0.11s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv2d_transpose_3x3_2x2
0.11s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv2d_transpose_3x3_2x2
0.11s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.MLP_not_autograph
0.11s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator_not_use_function
0.11s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_vanilla_rnn_8
0.11s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mean_2x2_Replicator_use_function
0.11s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_batch_norm_training_1x2x2x3
0.11s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.VectorQuantizerEMA_autograph
0.11s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_CrossReplicaBatchNorm_autograph
0.11s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv1DTranspose_not_autograph
0.11s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv1d_lstm_3x3_2x2
0.11s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_group_norm_2_1x3x4
0.11s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_GroupNorm_autograph
0.11s call     sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testModelForwards1
0.11s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_mlp_nobias_3x4x5_1x3
0.11s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.VectorQuantizerEMA_autograph
0.11s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_cross_replica_batch_norm_1x2x2x3_Replicator_use_function
0.11s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_vqvae_ema_train_Replicator
0.11s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_base_batch_norm_scale_offset_1x2x2x3_Replicator_use_function
0.11s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.MLP_not_autograph
0.11s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mean_2x2_Replicator_use_function
0.10s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_mlp_nobias_3x4x5_1x3_Replicator
0.10s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv3d_transpose_3x3_2x2
0.10s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_layer_norm_1_1x3_2
0.10s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_LayerNorm_autograph
0.10s call     sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testModelForwards0
0.10s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_lstm_1
0.10s call     sonnet/src/recurrent_test.py::ConvNDLSTMTest::testComputationAgainstNumPy4
0.10s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_lstm_1
0.10s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(Conv1DLSTM, static_unroll)_not_autograph
0.10s call     sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_falls_back_to_graph0
0.10s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.VectorQuantizer_autograph
0.10s call     sonnet/src/recurrent_test.py::ConvNDLSTMTest::testComputationAgainstNumPy5
0.10s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_CrossReplicaBatchNorm_not_autograph
0.10s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv2d_transpose_3x3_2x2
0.10s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(Conv3DLSTM, static_unroll)_not_autograph
0.10s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_instance_norm_1_1x3_2
0.10s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv1D_autograph
0.10s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_lstm_1_Replicator_use_function
0.10s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_nets.VectorQuantizerEMA_autograph
0.10s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vanilla_rnn_8_Replicator_use_function
0.10s call     sonnet/src/initializers_test.py::RandomUniformInitializerTest::testBatchAgnostic
0.10s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_InstanceNorm_autograph
0.10s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_lstm_8_projected_1
0.10s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_DepthwiseConv2D_autograph
0.10s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_base_batch_norm_scale_offset_1x2x2x3
0.10s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Conv3DTranspose
0.10s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_batch_norm_training_1x2x2x3
0.10s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_lstm_8_projected_1
0.10s call     sonnet/src/recurrent_test.py::LSTMTest::testComputationAgainstNumPy5
0.10s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_lstm_3x3_2x2_Replicator_not_use_function
0.10s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Conv2DTranspose
0.10s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(Conv2DLSTM, static_unroll)_not_autograph
0.10s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv1d_3x3_2x2
0.10s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv3D_autograph
0.10s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mlp_nobias_3x4x5_1x3_Replicator_use_function
0.10s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv2D_autograph
0.10s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_bias_3x3x3_Replicator_use_function
0.10s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Linear_autograph
0.09s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_cross_replica_batch_norm_1x2x2x3
0.09s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv3DTranspose_not_autograph
0.09s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_ema_eval_Replicator_use_function
0.09s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.VectorQuantizerEMA_not_autograph
0.09s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv2DTranspose_not_autograph
0.09s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_mean_2x2
0.09s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(VanillaRNN, static_unroll)_construct_module_in_function
0.09s call     sonnet/src/reshape_test.py::ReshapeTest::testRunSession1
0.09s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mlp_3x4x5_1x3_Replicator_use_function
0.09s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Conv1DTranspose
0.09s call     sonnet/src/initializers_test.py::RandomNormalInitializerTest::testBatchAgnostic
0.09s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_BatchNorm_not_autograph
0.09s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv1DTranspose_autograph
0.09s call     sonnet/src/conv_transpose_test.py::ConvTransposeTest::testGraphConv1
0.09s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_depthwise_conv2d_3x3_2x2
0.09s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(VanillaRNN, static_unroll)_not_construct_module_in_function
0.09s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_lstm_8_projected_1_Replicator_use_function
0.09s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.VectorQuantizerEMA_not_autograph
0.09s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv3d_3x3_2x2
0.09s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv2d_3x3_2x2
0.09s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(GRU, static_unroll)_not_autograph
0.09s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_linear_1x1
0.09s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(VanillaRNN, static_unroll)_not_autograph
0.09s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_nets.VectorQuantizer
0.09s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.VectorQuantizerEMA_not_construct_module_in_function
0.09s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.VectorQuantizer_not_autograph
0.09s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(VanillaRNN, static_unroll)_not_autograph
0.09s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_sum_2x2_Replicator_use_function
0.09s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_nets.VectorQuantizerEMA_not_autograph
0.09s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_training_1x2x2x3_Replicator_use_function
0.09s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_nets.VectorQuantizerEMA
0.08s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_sum_2x2
0.08s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_transpose_3x3_2x2_Replicator_use_function
0.08s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_nets.VectorQuantizerEMA
0.08s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_dropout
0.08s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_GroupNorm_not_autograph
0.08s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.Adam_not_autograph
0.08s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_trainable_state
0.08s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv3d_3x3_2x2
0.08s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv1d_transpose_3x3_2x2_Replicator
0.08s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_embed_100_100
0.08s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_Replicator_use_function
0.08s call     sonnet/src/initializers_test.py::OrthogonalInitializerTest::testTFFunction
0.08s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv3d_3x3_2x2
0.08s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_base_batch_norm_1x2x2x3
0.08s call     sonnet/src/utils_test.py::SmartAutographTest::test_smart_ag
0.08s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_ema_2_Replicator
0.08s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_bias_3x3x3
0.08s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vanilla_rnn_8_Replicator_use_function
0.08s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_transpose_3x3_2x2_Replicator_use_function
0.08s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_LayerNorm_not_autograph
0.08s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_Replicator_use_function
0.08s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_group_norm_2_1x3x4_Replicator_use_function
0.08s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_InstanceNorm_not_autograph
0.08s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv1D_not_autograph
0.08s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv3d_transpose_3x3_2x2_Replicator
0.08s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_GroupNorm
0.08s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.RMSProp_autograph
0.08s call     sonnet/src/base_test.py::BaseTest::testControlFlow
0.08s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_transpose_3x3_2x2_Replicator_use_function
0.08s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Conv2DTranspose
0.08s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Conv3DTranspose
0.08s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv2d_transpose_3x3_2x2_Replicator
0.08s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv1DTranspose_not_autograph
0.08s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator_not_use_function
0.08s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_depthwise_conv2d_3x3_2x2
0.08s call     sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_requires_eager0
0.08s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_embed_100_100_Replicator_use_function
0.08s call     sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_linear_nobias_1x1
0.08s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_depthwise_conv2d_3x3_2x2
0.08s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv2d_3x3_2x2
0.07s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_DepthwiseConv2D_not_autograph
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_layer_norm_1_1x3_2_Replicator_use_function
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator
0.07s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Bias_autograph
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_sum_2x2_Replicator_use_function
0.07s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Embed
0.07s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Conv1DTranspose
0.07s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_mlp_3x4x5_1x3
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_instance_norm_1_1x3_2_Replicator_use_function
0.07s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv3D_not_autograph
0.07s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv2D_not_autograph
0.07s call     sonnet/src/conv_transpose_test.py::ConvTransposeTest::testUnknownBatchSizeNHWC
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_3x3_2x2_Replicator_use_function
0.07s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv2DTranspose_autograph
0.07s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv3DTranspose_autograph
0.07s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_mlp_3x4x5_1x3
0.07s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Linear_not_autograph
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator_not_use_function
0.07s call     sonnet/src/conv_transpose_test.py::ConvTransposeTest::testUnknownShapeDims
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_gru_1_Replicator_use_function
0.07s call     sonnet/src/conformance/descriptors_test.py::DescriptorsTest::test_coverage
0.07s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_nets.MLP_autograph
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_transpose_3x3_2x2_Replicator_use_function
0.07s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_InstanceNorm
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_cross_replica_batch_norm_1x2x2x3_Replicator_use_function
0.07s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_vanilla_rnn_8
0.07s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Conv1D
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_transpose_3x3_2x2_Replicator_use_function
0.07s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_LayerNorm
0.07s call     sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testWithTFFunction0
0.07s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.Adam_not_autograph
0.07s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_cifar10_convnet_2x3_2x2_1x3x3x2
0.07s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_instance_norm_1_1x3_2_Replicator
0.07s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_layer_norm_1_1x3_2_Replicator
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mlp_nobias_3x4x5_1x3_Replicator_use_function
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_base_batch_norm_scale_offset_1x2x2x3_Replicator_use_function
0.07s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.RMSProp_not_autograph
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_depthwise_conv2d_3x3_2x2_Replicator_use_function
0.07s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv1DTranspose_autograph
0.07s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv1d_transpose_3x3_2x2
0.07s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Conv3DTranspose
0.07s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv2d_3x3_2x2_Replicator
0.07s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv3d_3x3_2x2_Replicator
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_3x3_2x2_Replicator_use_function
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_group_norm_2_1x3x4_Replicator_use_function
0.07s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Conv2DTranspose
0.07s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_3x3_2x2_Replicator_use_function
0.07s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv1d_transpose_3x3_2x2
0.06s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_cross_replica_batch_norm_1x2x2x3_Replicator
0.06s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_transpose_3x3_2x2_Replicator_use_function
0.06s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_lstm_3x3_2x2_Replicator_use_function
0.06s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_vanilla_rnn_8
0.06s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Conv1DTranspose
0.06s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv2DTranspose_not_autograph
0.06s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_cifar10_convnet_2x3_2x2_1x3x3x2
0.06s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_embed_100_100_Replicator_not_use_function
0.06s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_DepthwiseConv2D
0.06s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_linear_1x1_Replicator_use_function
0.06s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_mlp_nobias_3x4x5_1x3
0.06s call     sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testWithTFFunction1
0.06s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv3DTranspose_autograph
0.06s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_mlp_nobias_3x4x5_1x3
0.06s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Conv2D
0.06s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Conv3D
0.06s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_GroupNorm
0.06s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Linear
0.06s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv3DTranspose_not_autograph
0.06s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_GroupNorm
0.06s call     sonnet/src/linear_test.py::LinearTest::testMultiBatchOutputDimensions
0.06s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_ema_2
0.06s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv2DTranspose_autograph
0.06s call     sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testDefaultInitializers0
0.06s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_layer_norm_1_1x3_2_Replicator_use_function
0.06s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_instance_norm_1_1x3_2_Replicator_use_function
0.06s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_lstm_3x3_2x2_Replicator_use_function
0.06s call     sonnet/src/linear_test.py::LinearTest::testUnknownBatchSize
0.06s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_GroupNorm_autograph
0.06s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mlp_3x4x5_1x3_Replicator_use_function
0.06s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_GroupNorm_autograph
0.06s call     sonnet/src/custom_getter_test.py::DoctestTest::testDoctest
0.06s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Conv1D
0.06s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Conv3DTranspose
0.06s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_vqvae_ema_eval_Replicator
0.06s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Conv1D
0.06s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_InstanceNorm
0.06s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.VectorQuantizer_autograph
0.06s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Conv2DTranspose
0.06s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv1DTranspose_construct_module_in_function
0.06s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_InstanceNorm
0.06s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_lstm_3x3_2x2_Replicator_use_function
0.06s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv1DTranspose_not_construct_module_in_function
0.06s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv1DTranspose_not_autograph
0.06s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_lstm_8_projected_1_Replicator_use_function
0.06s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_3x3_2x2_Replicator_use_function
0.06s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_LayerNorm
0.06s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Conv2D
0.05s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_LayerNorm
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_3x3_2x2_Replicator_use_function
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_ema_train_Replicator_not_use_function
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_nets.VectorQuantizer_autograph
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_LayerNorm_autograph
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_lstm_3x3_2x2_Replicator_not_use_function
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.VectorQuantizer_not_autograph
0.05s call     sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testUnknownSpatialDims
0.05s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_mean_2x2_Replicator
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_lstm_1_Replicator_use_function
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Bias_not_autograph
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv1D_autograph
0.05s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_DepthwiseConv2D
0.05s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Conv3D
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv1D_autograph
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_LayerNorm_autograph
0.05s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_GroupNorm
0.05s call     sonnet/src/initializers_test.py::IdentityInitializerTest::testTFFunction4D
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.RMSProp_not_autograph
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_InstanceNorm_autograph
0.05s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Conv3D
0.05s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.VectorQuantizer_not_construct_module_in_function
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_InstanceNorm_autograph
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.VectorQuantizer_autograph
0.05s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Conv2D
0.05s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_DepthwiseConv2D
0.05s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Linear
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_GroupNorm_not_autograph
0.05s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_GroupNorm_construct_module_in_function
0.05s call     sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testFunctionSamePaddingUseBias
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_GroupNorm_not_autograph
0.05s call     sonnet/src/initializers_test.py::ConstantInitializersTest::testBatchAgnostic
0.05s call     sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testFunctionValidPaddingUseBias
0.05s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv2DTranspose_construct_module_in_function
0.05s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_GroupNorm_not_construct_module_in_function
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv2DTranspose_not_autograph
0.05s call     sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testResetWithTFFunction1
0.05s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Linear
0.05s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv3DTranspose_not_construct_module_in_function
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_unrolled_lstm_1_Replicator_not_use_function
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_DepthwiseConv2D_autograph
0.05s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv3DTranspose_construct_module_in_function
0.05s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Conv1D
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv3DTranspose_not_autograph
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_depthwise_conv2d_3x3_2x2_Replicator_use_function
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_DepthwiseConv2D_autograph
0.05s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv2DTranspose_not_construct_module_in_function
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_scale_offset_1x2x2x3_Replicator_use_function
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_3x3_2x2_Replicator_use_function
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.Momentum_autograph
0.05s call     sonnet/src/conv_test.py::ConvTest::testFunctionsamePaddingUseBias
0.05s call     sonnet/src/conv_test.py::ConvTest::testFunctionSamePaddingUseBias
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_1x2x2x3_Replicator_use_function
0.05s call     sonnet/src/conv_transpose_test.py::ConvTransposeTest::testGraphConv3
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.VectorQuantizer_not_autograph
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_nets.VectorQuantizer_not_autograph
0.05s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_cifar10_convnet_2x3_2x2_1x3x3x2
0.05s call     sonnet/src/conv_test.py::ConvTest::testUnknownSpatialDims
0.05s call     sonnet/src/conv_test.py::ConvTest::testFunctionvalidPaddingUseBias
0.05s call     sonnet/src/conv_test.py::ConvTest::testFunctionValidPaddingUseBias
0.05s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_InstanceNorm
0.05s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_vqvae_Replicator
0.05s call     sonnet/src/nets/vqvae_test.py::VqvaeTest::testEmaUpdating2
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_InstanceNorm_not_autograph
0.05s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv1d_3x3_2x2
0.05s call     sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testUnknownBatchSizeNHWC
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_lstm_1_Replicator_not_use_function
0.05s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction0
0.05s call     sonnet/src/nets/vqvae_test.py::VqvaeTest::testEmaUpdating3
0.05s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_LayerNorm
0.05s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_dropout
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv2D_autograph
0.05s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_nets.VectorQuantizer
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv3D_autograph
0.05s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_InstanceNorm_construct_module_in_function
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_linear_nobias_1x1_Replicator_use_function
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_embed_100_100_Replicator_use_function
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_ema_2_Replicator_use_function
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(VanillaRNN, static_unroll)_not_autograph
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Linear_autograph
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv2D_autograph
0.05s call     sonnet/src/conv_transpose_test.py::ConvTransposeTest::testGraphConv2
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv3D_autograph
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Linear_autograph
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_training_1x2x2x3_Replicator_not_use_function
0.05s call     sonnet/src/conv_test.py::ConvTest::testUnknownBatchSizeNHWC
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_LayerNorm_not_autograph
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_gru_1_Replicator_not_use_function
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_InstanceNorm_not_autograph
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_linear_1x1_Replicator_use_function
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_LayerNorm_not_autograph
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Conv1DTranspose_autograph
0.05s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_LayerNorm_construct_module_in_function
0.05s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_InstanceNorm_not_construct_module_in_function
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv1D_not_autograph
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vanilla_rnn_8_Replicator_use_function
0.05s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_LayerNorm_not_construct_module_in_function
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_bias_3x3x3_Replicator_use_function
0.05s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_nets.VectorQuantizer
0.05s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv1D_not_autograph
0.05s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv1D_construct_module_in_function
0.05s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_DepthwiseConv2D
0.05s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_ema_2
0.05s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_transpose_3x3_2x2_Replicator_use_function
0.05s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv1D_not_construct_module_in_function
0.05s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Conv2D
0.04s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Conv3D
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_embed_100_100
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_ema_eval_Replicator_not_use_function
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_transpose_3x3_2x2_Replicator_use_function
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_group_norm_2_1x3x4
0.04s call     sonnet/src/initializers_test.py::RandomNormalInitializerTest::testTFFunction
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_ema_train_Replicator_not_use_function
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_embed_100_100
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_GroupNorm_autograph
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_group_norm_2_1x3x4
0.04s call     sonnet/src/linear_test.py::LinearTest::testFunction0
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_DepthwiseConv2D_not_autograph
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv3D_not_autograph
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_batch_norm_1x2x2x3
0.04s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Linear
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_transpose_3x3_2x2_Replicator_use_function
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv3D_not_autograph
0.04s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_DepthwiseConv2D_construct_module_in_function
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_DepthwiseConv2D_not_autograph
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.Momentum_not_autograph
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_mean_2x2
0.04s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv3D_construct_module_in_function
0.04s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_DepthwiseConv2D_not_construct_module_in_function
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv1d_3x3_2x2
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_batch_norm_1x2x2x3
0.04s call     sonnet/src/initializers_test.py::IdentityInitializerTest::testTFFunction
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_instance_norm_1_1x3_2_Replicator_use_function
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_scale_offset_1x2x2x3_Replicator_not_use_function
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_1x2x2x3_Replicator_not_use_function
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_layer_norm_1_1x3_2
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_batch_norm_scale_offset_1x2x2x3
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_instance_norm_1_1x3_2
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_layer_norm_1_1x3_2_Replicator_use_function
0.04s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_bias_3x3x3_Replicator
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_layer_norm_1_1x3_2
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv2D_not_autograph
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_instance_norm_1_1x3_2
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Conv3DTranspose_autograph
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv2D_not_autograph
0.04s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv2D_not_construct_module_in_function
0.04s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv2D_construct_module_in_function
0.04s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv3D_not_construct_module_in_function
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_linear_1x1_Replicator_use_function
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Linear_not_autograph
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mlp_3x4x5_1x3_Replicator_not_use_function
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_cross_replica_batch_norm_1x2x2x3
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Linear_not_autograph
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Conv2DTranspose_autograph
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_training_1x2x2x3_Replicator_not_use_function
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Flatten_autograph
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_base_batch_norm_scale_offset_1x2x2x3
0.04s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Linear_construct_module_in_function
0.04s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Linear_not_construct_module_in_function
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_3x3_2x2_Replicator_use_function
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_lstm_3x3_2x2_Replicator_not_use_function
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Dropout_autograph
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_InstanceNorm_autograph
0.04s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_LayerNorm_autograph
0.04s call     sonnet/src/initializers_test.py::TruncatedNormalInitializerTest::testTFFunction
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_linear_nobias_1x1_Replicator_use_function
0.04s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Bias
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_3x3_2x2_Replicator_use_function
0.04s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_cifar10_convnet_2x3_2x2_1x3x3x2
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mean_2x2_Replicator_use_function
0.04s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_cifar10_convnet_2x3_2x2_1x3x3x2
0.04s call     sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testDefaultInitializers1
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_base_batch_norm_scale_offset_1x2x2x3_Replicator_use_function
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_cross_replica_batch_norm_1x2x2x3_Replicator_use_function
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_ema_eval_Replicator_not_use_function
0.04s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_vqvae_ema_train
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_linear_1x1
0.04s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_depthwise_conv2d_3x3_2x2_Replicator_use_function
0.04s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_mean_2x2
0.04s call     sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testFunctionValidPaddingNoBias
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_3x3_2x2_Replicator_use_function
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Embed_autograph
0.03s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_linear_1x1
0.03s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_vqvae_ema_train
0.03s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_sum_2x2
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_vqvae_ema_train_Replicator
0.03s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_cifar10_convnet_2x3_2x2_1x3x3x2
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_batch_norm_training_1x2x2x3_Replicator
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_nets.MLP_not_autograph
0.03s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_batch_norm_training_1x2x2x3
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_lstm_8_projected_1_Replicator_not_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_1x2x2x3_Replicator_not_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_scale_offset_1x2x2x3_Replicator_not_use_function
0.03s call     sonnet/src/conv_test.py::ConvTest::testFunctionsamePaddingNoBias
0.03s call     sonnet/src/conv_test.py::ConvTest::testFunctionValidPaddingNoBias
0.03s call     sonnet/src/conv_test.py::ConvTest::testFunctionvalidPaddingNoBias
0.03s call     sonnet/src/initializers_test.py::TruncatedNormalInitializerTest::testBatchAgnostic
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Conv1D_autograph
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_GroupNorm_not_autograph
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mlp_3x4x5_1x3_Replicator_not_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_dropout_Replicator_use_function
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Sequential_autograph
0.03s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_instance_norm_1_1x3_2
0.03s call     sonnet/src/initializers_test.py::ConstantInitializersTest::testTFFunction
0.03s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_bias_3x3x3
0.03s call     sonnet/src/recurrent_test.py::TrainableStateTest::testMasked
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Conv1DTranspose_not_autograph
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_ema_train_Replicator_not_use_function
0.03s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_bias_3x3x3
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vanilla_rnn_8_Replicator_not_use_function
0.03s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv2d_3x3_2x2
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_trainable_state_Replicator_use_function
0.03s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Bias
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_training_1x2x2x3_Replicator_not_use_function
0.03s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_sum_2x2
0.03s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_batch_norm_training_1x2x2x3
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_1x2x2x3_Replicator_not_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_bias_3x3x3_Replicator_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_batch_norm_1x2x2x3_Replicator
0.03s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Bias
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_scale_offset_1x2x2x3_Replicator_not_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_lstm_3x3_2x2_Replicator_not_use_function
0.03s call     sonnet/src/initializers_test.py::RandomUniformInitializerTest::testTFFunction
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Flatten_not_autograph
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_batch_norm_scale_offset_1x2x2x3_Replicator
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_ema_eval_Replicator_not_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_transpose_3x3_2x2_Replicator_not_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_gru_1_Replicator_not_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_vqvae_ema_eval_Replicator
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Dropout_not_autograph
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_trainable_state_Replicator_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_vqvae_ema_eval
0.03s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Embed_construct_module_in_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mlp_3x4x5_1x3_Replicator_not_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_mlp_3x4x5_1x3_Replicator
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.Momentum_not_autograph
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_linear_nobias_1x1_Replicator_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_unrolled_lstm_1_Replicator_not_use_function
0.03s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_trainable_state
0.03s call     sonnet/src/linear_test.py::LinearTest::testFunction1
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_lstm_3x3_2x2_Replicator_not_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_vqvae_ema_eval
0.03s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_batch_norm_1x2x2x3
0.03s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_batch_norm_scale_offset_1x2x2x3
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_3x3_2x2_Replicator_not_use_function
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_DepthwiseConv2D_autograph
0.03s call     sonnet/src/dropout_test.py::DropoutTest::test_dropout_is_actually_random
0.03s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_trainable_state
0.03s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_vqvae_ema_train
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_LayerNorm_not_autograph
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_InstanceNorm_not_autograph
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_lstm_8_projected_1_Replicator_not_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_lstm_3x3_2x2_Replicator_not_use_function
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_embed_100_100_Replicator_use_function
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Embed_not_autograph
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mlp_nobias_3x4x5_1x3_Replicator_not_use_function
0.03s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_batch_norm_scale_offset_1x2x2x3
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Conv2D_autograph
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Conv3D_autograph
0.03s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_batch_norm_1x2x2x3
0.03s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_mlp_3x4x5_1x3
0.03s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_batch_norm_scale_offset_1x2x2x3
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_lstm_1_Replicator_not_use_function
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Bias_autograph
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Sequential_not_autograph
0.03s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Bias_autograph
0.03s call     sonnet/src/dropout_test.py::DropoutTest::test_with_tf_function_with_variables0
0.03s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Flatten
0.03s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv2d_lstm_3x3_2x2
0.03s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_gru_1
0.03s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv2d_lstm_3x3_2x2_Replicator
0.03s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_linear_nobias_1x1
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vanilla_rnn_8_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_unrolled_lstm_1_Replicator
0.02s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Conv3DTranspose_not_autograph
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_gru_1_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_linear_1x1_Replicator_not_use_function
0.02s call     sonnet/src/dropout_test.py::DropoutTest::test_with_tf_function_with_variables1
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv1d_lstm_3x3_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_unrolled_lstm_1
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_batch_norm_training_1x2x2x3
0.02s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_cross_replica_batch_norm_1x2x2x3
0.02s call     sonnet/src/conformance/xla_test.py::XLATest::test_compile_base_batch_norm_scale_offset_1x2x2x3
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_mlp_3x4x5_1x3
0.02s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Conv2DTranspose_not_autograph
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv1d_lstm_3x3_2x2_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_lstm_8_projected_1_Replicator_not_use_function
0.02s call     sonnet/src/optimizers/adam_test.py::AdamTest::testSparse
0.02s call     sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_linear_nobias_1x1
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic6
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_lstm_8_projected_1_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_vqvae_ema_eval
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_lstm_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mlp_nobias_3x4x5_1x3_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_lstm_1_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_transpose_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_transpose_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_lstm_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Bias_not_autograph
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_ema_2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_embed_100_100_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_lstm_8_projected_1
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_Replicator_not_use_function
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic7
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_gru_1_Replicator_not_use_function
0.02s call     sonnet/src/group_norm_test.py::GroupNormTest::testWorksWithFunction
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_lstm_1
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv3d_lstm_3x3_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_lstm_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic8
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_unrolled_lstm_1_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv3d_lstm_3x3_2x2_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_ema_2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_transpose_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv1d_lstm_3x3_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_group_norm_2_1x3x4_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vanilla_rnn_8_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_group_norm_2_1x3x4_Replicator_not_use_function
0.02s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Bias_not_autograph
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_batch_norm_1x2x2x3
0.02s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Bias_autograph
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_batch_norm_scale_offset_1x2x2x3
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_gru_1
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_transpose_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_transpose_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_vqvae_ema_train
0.02s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Conv1D_not_autograph
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_sum_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_base_batch_norm_scale_offset_1x2x2x3_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_lstm_8_projected_1
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv3d_lstm_3x3_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_depthwise_conv2d_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Dropout
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic3
0.02s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_vqvae_ema_train
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_vanilla_rnn_8_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_sum_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_base_batch_norm_1x2x2x3_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mlp_nobias_3x4x5_1x3_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_mlp_nobias_3x4x5_1x3_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv2d_lstm_3x3_2x2
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction6
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_unrolled_lstm_1
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_base_batch_norm_scale_offset_1x2x2x3_Replicator_not_use_function
0.02s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Bias_construct_module_in_function
0.02s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Bias_not_construct_module_in_function
0.02s call     sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testModelCreation
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic4
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_lstm_1_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_depthwise_conv2d_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_cross_replica_batch_norm_1x2x2x3_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_base_batch_norm_1x2x2x3_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_instance_norm_1_1x3_2_Replicator_not_use_function
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic1
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_mlp_3x4x5_1x3
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_cross_replica_batch_norm_1x2x2x3_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_layer_norm_1_1x3_2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_dropout_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_vqvae_ema_train_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_linear_1x1_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_sum_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_instance_norm_1_1x3_2_Replicator_not_use_function
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction7
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_vanilla_rnn_8
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_layer_norm_1_1x3_2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_mlp_nobias_3x4x5_1x3
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_sum_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_base_batch_norm_1x2x2x3
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction3
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic2
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_vqvae
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction8
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_mlp_nobias_3x4x5_1x3
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mean_2x2_Replicator_not_use_function
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic5
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_batch_norm_training_1x2x2x3_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_cross_replica_batch_norm_1x2x2x3_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv3d_transpose_3x3_2x2_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_lstm_1
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_vanilla_rnn_8
0.02s call     sonnet/src/axis_norm_test.py::LayerNormTest::testWorksWithFunction
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_transpose_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv3d_transpose_3x3_2x2
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction1
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_ema_2_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_sum_2x2_Replicator
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction4
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mean_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_vqvae_ema_train
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_group_norm_2_1x3x4
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_vqvae_ema_eval_Replicator
0.02s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_embed_100_100_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_ema_2
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_group_norm_2_1x3x4_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_Replicator_not_use_function
0.02s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_batch_norm_training_1x2x2x3
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv1d_lstm_3x3_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_ema_2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_base_batch_norm_scale_offset_1x2x2x3_Replicator_not_use_function
0.02s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Sequential
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_vqvae_Replicator
0.02s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_DepthwiseConv2D_not_autograph
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_gru_1
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_vqvae
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_bias_3x3x3_Replicator_not_use_function
0.02s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Embed
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv1d_transpose_3x3_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_group_norm_2_1x3x4_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_base_batch_norm_scale_offset_1x2x2x3_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_batch_norm_1x2x2x3_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_transpose_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction2
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_base_batch_norm_1x2x2x3_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv2d_transpose_3x3_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_sum_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_transpose_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_batch_norm_scale_offset_1x2x2x3_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_lstm_8_projected_1
0.02s call     sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction5
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv1d_transpose_3x3_2x2_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv2d_transpose_3x3_2x2_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv3d_3x3_2x2_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_base_batch_norm_1x2x2x3_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_batch_norm_training_1x2x2x3
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv3d_lstm_3x3_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_ema_2
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_depthwise_conv2d_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_instance_norm_1_1x3_2_Replicator_not_use_function
0.02s call     sonnet/src/reshape_test.py::ReshapeTest::testRunSession8
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_instance_norm_1_1x3_2_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_group_norm_2_1x3x4
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_base_batch_norm_scale_offset_1x2x2x3
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_mlp_3x4x5_1x3_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_cross_replica_batch_norm_1x2x2x3_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_unrolled_lstm_1
0.02s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_batch_norm_training_1x2x2x3
0.02s call     sonnet/src/reshape_test.py::ReshapeTest::testRunSession3
0.02s call     sonnet/src/reshape_test.py::ReshapeTest::testRunSession5
0.02s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_vqvae_ema_eval
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv2d_lstm_3x3_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv1d_3x3_2x2_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv3d_3x3_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_layer_norm_1_1x3_2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_depthwise_conv2d_3x3_2x2_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_vqvae_ema_eval
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_cross_replica_batch_norm_1x2x2x3
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_depthwise_conv2d_3x3_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv1d_3x3_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_bias_3x3x3_Replicator_not_use_function
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_3x3_2x2_Replicator_not_use_function
0.02s call     sonnet/src/reshape_test.py::ReshapeTest::testRunSession9
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_layer_norm_1_1x3_2_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv2d_3x3_2x2_Replicator
0.02s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv1d_transpose_3x3_2x2
0.02s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv2d_lstm_3x3_2x2_Replicator
0.02s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_vqvae_ema_eval
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_layer_norm_1_1x3_2
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_trainable_state_Replicator_not_use_function
0.01s call     sonnet/src/reshape_test.py::FlattenTest::testFlatten_unknownBatchSize
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_batch_norm_1x2x2x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_cross_replica_batch_norm_1x2x2x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv2d_transpose_3x3_2x2
0.01s call     sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape2
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_linear_1x1_Replicator_not_use_function
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv3d_transpose_3x3_2x2
0.01s call     sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape1
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_batch_norm_scale_offset_1x2x2x3
0.01s call     sonnet/src/reshape_test.py::ReshapeTest::testRunSession6
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_base_batch_norm_1x2x2x3
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Conv2D_not_autograph
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Conv3D_not_autograph
0.01s call     sonnet/src/reshape_test.py::ReshapeTest::testRunSession4
0.01s call     sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape2
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_unrolled_lstm_1_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_linear_1x1_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_base_batch_norm_scale_offset_1x2x2x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_trainable_state_Replicator_not_use_function
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv2d_lstm_3x3_2x2
0.01s call     sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape3
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv2d_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mean_2x2_Replicator_not_use_function
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_linear_nobias_1x1_Replicator_not_use_function
0.01s call     sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape3
0.01s call     sonnet/src/dropout_test.py::DropoutTest::test_with_tf_function_with_booleans0
0.01s call     sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape1
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Flatten_autograph
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_layer_norm_1_1x3_2
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv2d_lstm_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_gru_1_Replicator
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_gru_1
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Linear_not_autograph
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_instance_norm_1_1x3_2
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv1d_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_lstm_1
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_vanilla_rnn_8
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_lstm_1_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_depthwise_conv2d_3x3_2x2
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv2d_lstm_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_mlp_nobias_3x4x5_1x3
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Dropout_autograph
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_mean_2x2_Replicator
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_unrolled_lstm_1
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_mlp_3x4x5_1x3
0.01s call     sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testSparseCentered
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_unrolled_lstm_1
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_dropout_Replicator_not_use_function
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_vqvae
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_linear_nobias_1x1_Replicator_not_use_function
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_gru_1
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv2d_3x3_2x2
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_gru_1
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_unrolled_lstm_1
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv3d_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv1d_lstm_3x3_2x2_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_sum_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_linear_1x1
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_batch_norm_1x2x2x3
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_batch_norm_scale_offset_1x2x2x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_linear_1x1
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv1d_lstm_3x3_2x2
0.01s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Flatten
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_ema_2
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_mean_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_trainable_state_Replicator_not_use_function
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_dropout_Replicator_not_use_function
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_mean_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_lstm_8_projected_1_Replicator
0.01s teardown sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_session
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_lstm_1
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_dropout_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_group_norm_2_1x3x4
0.01s call     sonnet/src/dropout_test.py::DropoutTest::test_with_tf_function_with_booleans1
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_dropout_Replicator_use_function
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_embed_100_100_Replicator_not_use_function
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_embed_100_100
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_trainable_state_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_embed_100_100_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv1d_lstm_3x3_2x2
0.01s call     sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testDenseCentered
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv1d_lstm_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv2d_transpose_3x3_2x2
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_mlp_3x4x5_1x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv1d_transpose_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_dropout_Replicator_not_use_function
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_lstm_1
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_batch_norm_1x2x2x3
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_batch_norm_scale_offset_1x2x2x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv3d_transpose_3x3_2x2
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_embed_100_100
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_trainable_state
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_layer_norm_1_1x3_2
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_trainable_state
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_lstm_8_projected_1
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_bias_3x3x3_Replicator_not_use_function
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_mlp_3x4x5_1x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_cross_replica_batch_norm_1x2x2x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_linear_nobias_1x1_Replicator_not_use_function
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_base_batch_norm_scale_offset_1x2x2x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_base_batch_norm_1x2x2x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_instance_norm_1_1x3_2
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv1d_3x3_2x2
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_lstm_1
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_embed_100_100
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_bias_3x3x3_Replicator
0.01s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Dropout
0.01s call     sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testSparse
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Embed_autograph
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv3d_lstm_3x3_2x2_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_dropout
0.01s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Dropout
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Flatten_autograph
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_linear_nobias_1x1_Replicator
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Flatten_not_autograph
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_depthwise_conv2d_3x3_2x2
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_embed_100_100
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_vanilla_rnn_8_Replicator
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Dropout_autograph
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Dropout_autograph
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_mlp_nobias_3x4x5_1x3_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_bias_3x3x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_bias_3x3x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv3d_3x3_2x2
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.SGD_autograph
0.01s call     sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Sequential
0.01s call     sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testSparseCentered
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv2d_3x3_2x2
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_lstm_8_projected_1
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_linear_1x1
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_linear_nobias_1x1
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_linear_nobias_1x1
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv3d_lstm_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_mean_2x2
0.01s call     sonnet/src/reshape_test.py::ReshapeTest::testPreservePreserve?5
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Bias_not_autograph
0.01s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Embed
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_dropout
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Dropout_not_autograph
0.01s call     sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_activate_final0
0.01s call     sonnet/src/reshape_test.py::ReshapeTest::testUnknownBatchSize
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv3d_lstm_3x3_2x2
0.01s call     sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testDense
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_lstm_8_projected_1
0.01s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Embed
0.01s call     sonnet/src/reshape_test.py::ReshapeTest::testPreservePreserve5?
0.01s call     sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_activate_final1
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_vanilla_rnn_8
0.01s call     sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_activation
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Dropout_not_autograph
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Dropout_not_autograph
0.01s call     sonnet/src/base_test.py::TestModuleNaming::test_module_numbering_in_graph
0.01s call     sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape_has_static_info_in_graph2
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv3d_transpose_3x3_2x2_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_mlp_nobias_3x4x5_1x3
0.01s call     sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape_has_static_info_in_graph2
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv3d_lstm_3x3_2x2
0.01s call     sonnet/src/reshape_test.py::ReshapeTest::testPreservePreserve??
0.01s call     sonnet/src/reshape_test.py::ReshapeTest::testPreservePreserve?
0.01s call     sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape_has_static_info_in_graph3
0.01s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Flatten
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_vqvae
0.01s call     sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape_has_static_info_in_graph1
0.01s call     sonnet/src/reshape_test.py::ReshapeTest::testRunSession2
0.01s call     sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape_has_static_info_in_graph3
0.01s call     sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape_has_static_info_in_graph1
0.01s call     sonnet/src/batch_norm_test.py::BaseBatchNormTest::testUsingTestStats
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_vqvae
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_trainable_state
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv3d_transpose_3x3_2x2
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.SGD_not_autograph
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv3d_3x3_2x2_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_embed_100_100
0.01s call     sonnet/src/optimizers/adam_test.py::AdamTest::testDense
0.01s call     sonnet/src/recurrent_test.py::UnrolledLSTMTest::testComputationAgainstLSTM3
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_base_batch_norm_1x2x2x3_Replicator
0.01s call     sonnet/src/reshape_test.py::ReshapeTest::testRunSession7
0.01s call     sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testDenseCentered
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_ema_2_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_sum_2x2
0.01s call     sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testSparse
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_vanilla_rnn_8
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_base_batch_norm_1x2x2x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv2d_transpose_3x3_2x2_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_group_norm_2_1x3x4_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_dropout
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_vanilla_rnn_8
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Embed_autograph
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv1d_transpose_3x3_2x2_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_bias_3x3x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_base_batch_norm_scale_offset_1x2x2x3_Replicator
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_mlp_nobias_3x4x5_1x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv2d_3x3_2x2_Replicator
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Embed_autograph
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_cross_replica_batch_norm_1x2x2x3_Replicator
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Flatten_autograph
0.01s call     sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testSparseNesterov
0.01s call     sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testSparse
0.01s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Dropout_construct_module_in_function
0.01s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Dropout_not_construct_module_in_function
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_vqvae_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv3d_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_depthwise_conv2d_3x3_2x2_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_linear_nobias_1x1
0.01s call     sonnet/src/reshape_test.py::FlattenTest::testFlatten_unknownNonBatchSize
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv1d_3x3_2x2_Replicator
0.01s call     sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testVariableHyperParams
0.01s call     sonnet/src/nets/dnc/control_test.py::DeepCore::testShape3
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.SGD_autograph
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv3d_transpose_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv2d_transpose_3x3_2x2
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_mlp_nobias_3x4x5_1x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_layer_norm_1_1x3_2_Replicator
0.01s call     sonnet/src/optimizers/adam_test.py::AdamTest::testVariableHyperParams
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_ema_2
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv1d_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_base_batch_norm_scale_offset_1x2x2x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_group_norm_2_1x3x4
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv1d_transpose_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_vqvae
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv3d_transpose_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_instance_norm_1_1x3_2_Replicator
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Embed_not_autograph
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_mean_2x2_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_cross_replica_batch_norm_1x2x2x3
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Embed_not_autograph
0.01s teardown sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(UnrolledLSTM)_autograph
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Embed_not_autograph
0.01s call     sonnet/src/recurrent_test.py::UnrolledLSTMTest::testComputationAgainstLSTM1
0.01s call     sonnet/src/recurrent_test.py::LSTMTest::testRecurrentDropout0
0.01s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Dropout
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_linear_1x1_Replicator
0.01s call     sonnet/src/recurrent_test.py::ConvNDLSTMTest::testComputationAgainstNumPy0
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv2d_3x3_2x2
0.01s call     sonnet/src/conformance/doctest_test.py::DoctestTest::test_doctest_sonnet.initializers
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_group_norm_2_1x3x4
0.01s call     sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthRange0
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_depthwise_conv2d_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_embed_100_100_Replicator
0.01s call     sonnet/src/optimizers/adam_test.py::AdamTest::testHyperParamDTypeConversion
0.01s call     sonnet/src/recurrent_test.py::LSTMTest::testRecurrentDropout1
0.01s call     sonnet/src/recurrent_test.py::LSTMTest::testRecurrentDropout2
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv1d_transpose_3x3_2x2
0.01s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv1DLSTM_dynamic_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_instance_norm_1_1x3_2
0.01s call     sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Embed_not_construct_module_in_function
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_group_norm_2_1x3x4
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_layer_norm_1_1x3_2
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Sequential_autograph
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_mean_2x2
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv2d_transpose_3x3_2x2
0.01s call     sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testDense
0.01s call     sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_b_init
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv3d_3x3_2x2
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_linear_1x1
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_ema_2
0.01s teardown sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Recurrent(UnrolledLSTM)_not_autograph
0.01s call     sonnet/src/batch_norm_test.py::BatchNormTest::testSimple
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_sum_2x2_Replicator
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_base_batch_norm_1x2x2x3
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv2d_transpose_3x3_2x2
0.01s call     sonnet/src/recurrent_test.py::DeepRNNTest::testComputationAgainstNumPy0
0.01s call     sonnet/src/recurrent_test.py::DeepRNNTest::testComputationAgainstNumPyWithCallables1
0.01s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv1DLSTM_not_dynamic_Replicator
0.01s call     sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testVariableHyperParams
0.01s call     sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testDense
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv3d_3x3_2x2
0.01s call     sonnet/src/nets/vqvae_test.py::VqvaeTest::testConstruct1
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.SGD_not_autograph
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv1d_transpose_3x3_2x2
0.01s call     sonnet/src/optimizers/momentum_test.py::MomentumTest::testSparseNesterov
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_base_batch_norm_scale_offset_1x2x2x3
0.01s call     sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testHyperParamDTypeConversion
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv1d_3x3_2x2
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_ema_2
0.01s call     sonnet/src/optimizers/momentum_test.py::MomentumTest::testSparse
0.01s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_GRU_not_dynamic_Replicator
0.01s call     sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testDenseNesterov
0.01s call     sonnet/src/parallel_linear_test.py::ParallelLinearTest::test_behaves_same_as_stacked_linears
0.01s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_GRU_dynamic_Replicator
0.01s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv3DLSTM_dynamic_Replicator
0.01s call     sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_passed_with_bias0
0.01s call     sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_w_init
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_cross_replica_batch_norm_1x2x2x3
0.01s call     sonnet/src/nets/mlp_test.py::MLPTest::test_reverse
0.01s call     sonnet/src/nets/vqvae_test.py::VqvaeTest::testConstruct0
0.01s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv2DLSTM_dynamic_Replicator
0.01s call     sonnet/src/nets/dnc/control_test.py::DeepCore::testShape1
0.01s call     sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthOneZeroLength0
0.01s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv3DLSTM_not_dynamic_Replicator
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_trainable_state_Replicator
0.01s call     sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_custom_name
0.01s call     sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_override_name
0.01s call     sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_default_name
0.01s call     sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthAllFull0
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_dropout_Replicator
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_instance_norm_1_1x3_2
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_depthwise_conv2d_3x3_2x2
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_base_batch_norm_1x2x2x3
0.01s call     sonnet/src/conformance/build_test.py::BuildTest::test_build_Flatten
0.01s call     sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthRange1
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_layer_norm_1_1x3_2
0.01s call     sonnet/src/recurrent_test.py::UnrolledLSTMTest::testNumStepsPolymorphism1
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_cross_replica_batch_norm_1x2x2x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_bias_3x3x3_Replicator
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_base_batch_norm_scale_offset_1x2x2x3
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_embed_100_100
0.01s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv2DLSTM_not_dynamic_Replicator
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_linear_1x1
0.01s call     sonnet/src/recurrent_test.py::DeepRNNTest::testWithSkipConnectionsOutputs0
0.01s call     sonnet/src/conv_test.py::ConvTest::testDefaultInitializers1
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv2d_3x3_2x2
0.01s call     sonnet/src/conformance/doctest_test.py::DoctestTest::test_doctest_sonnet.mixed_precision
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv1d_3x3_2x2
0.01s call     sonnet/src/reshape_test.py::ReshapeTest::testPreservePreserve1
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Flatten_not_autograph
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_trainable_state
0.01s call     sonnet/src/nets/vqvae_test.py::VqvaeTest::testNoneBatch1
0.01s call     sonnet/src/conformance/function_test.py::FunctionTest::test_create_variables_eagerly_Flatten_not_autograph
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_depthwise_conv2d_3x3_2x2
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_instance_norm_1_1x3_2
0.01s call     sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_linear_nobias_1x1_Replicator
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_layer_norm_1_1x3_2
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_linear_1x1
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_dropout
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv2d_3x3_2x2
0.01s call     sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Sequential
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_bias_3x3x3
0.01s call     sonnet/src/conformance/doctest_test.py::DoctestTest::test_doctest_sonnet.regularizers
0.01s call     sonnet/src/reshape_test.py::ReshapeTest::testPreservePreserve24
0.01s call     sonnet/src/recurrent_test.py::ConvNDLSTMTest::testComputationAgainstNumPy2
0.01s call     sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Sequential
0.01s call     sonnet/src/recurrent_test.py::ConvNDLSTMTest::testComputationAgainstNumPy1
0.01s call     sonnet/src/conv_test.py::ConvTest::testDefaultInitializers0
0.01s call     sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_linear_nobias_1x1
0.01s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_VanillaRNN_dynamic_Replicator
0.01s call     sonnet/src/recurrent_test.py::ConvNDLSTMTest::testInitialization
0.01s call     sonnet/src/nets/mlp_test.py::MLPTest::test_applies_activation_with_dropout3
0.01s call     sonnet/src/recurrent_test.py::UnrollTest::testNestedOutputs0
0.01s call     sonnet/src/optimizers/adam_test.py::ReferenceAdamTest::testNoneUpdate
0.01s call     sonnet/src/batch_norm_test.py::BaseBatchNormTest::testSimpleTraining3DNCDHW
0.01s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_LSTM_dynamic_Replicator
0.01s call     sonnet/src/batch_norm_test.py::BaseBatchNormTest::testSimpleTraining3D
0.01s call     sonnet/src/optimizers/momentum_test.py::MomentumTest::testVariableHyperParams
0.01s call     sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testNoneUpdate
0.01s call     sonnet/src/nets/vqvae_test.py::VqvaeTest::testShapeChecking1
0.01s call     sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthOneZeroLength1
0.01s call     sonnet/src/batch_norm_test.py::BaseBatchNormTest::testNoScaleAndOffset
0.01s call     sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_mean_2x2
0.01s call     sonnet/src/recurrent_test.py::GRUTest::testComputationAgainstNumPy0
0.01s call     sonnet/src/group_norm_test.py::GroupNormTest::testDataFormatAgnosticVar
0.01s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_LSTM_not_dynamic_Replicator
0.01s call     sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_VanillaRNN_not_dynamic_Replicator
0.01s call     sonnet/src/recurrent_test.py::LSTMTest::testComputationAgainstNumPy2
0.01s call     sonnet/src/recurrent_test.py::LSTMTest::testComputationAgainstNumPy3
0.01s call     sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_passed_with_bias1
0.01s call     sonnet/src/group_norm_test.py::GroupNormTest::testSliceAxis
0.01s call     sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testHyperParamDTypeConversion
0.01s call     sonnet/src/nets/dnc/control_test.py::DeepCore::testShape2
0.01s call     sonnet/src/optimizers/sgd_test.py::ReferenceSGDTest::testVariableLearningRate
0.01s call     sonnet/src/conformance/copy_test.py::CopyTest::test_copy_mean_2x2
0.01s call     sonnet/src/recurrent_test.py::UnrollTest::testNestedInputs0
0.01s call     sonnet/src/optimizers/momentum_test.py::MomentumTest::testDenseNesterov

(0.00 durations hidden.  Use -vv to show these durations.)
=========================== short test summary info ============================
FAILED sonnet/src/conv_test.py::ConvTest::testPaddingFunctionReached - Attrib...
=== 1 failed, 2464 passed, 552 skipped, 10665 warnings in 379.81s (0:06:19) ====
