docs/ext/link_tf_api_test.py::LinkTfApiTest::test_link_to_method_of_exported_class
docs/ext/link_tf_api_test.py::LinkTfApiTest::test_link_to_nested_package
docs/ext/link_tf_api_test.py::LinkTfApiTest::test_link_to_non_existent_method_of_exported_class
docs/ext/link_tf_api_test.py::LinkTfApiTest::test_link_to_top_level
docs/ext/link_tf_api_test.py::LinkTfApiTest::test_non_existent
examples/simple_mnist_test.py::SimpleMnistTest::test_session
examples/simple_mnist_test.py::SimpleMnistTest::test_test_accuracy
examples/simple_mnist_test.py::SimpleMnistTest::test_train_epoch
sonnet/src/axis_norm_test.py::LayerNormTest::test3DDataFormatAgnostic
sonnet/src/axis_norm_test.py::LayerNormTest::test5DDataFormatAgnostic
sonnet/src/axis_norm_test.py::LayerNormTest::testCreateOffsetAndOffsetProvided
sonnet/src/axis_norm_test.py::LayerNormTest::testCreateScaleAndScaleProvided
sonnet/src/axis_norm_test.py::LayerNormTest::testDataFormatAgnosticTensor
sonnet/src/axis_norm_test.py::LayerNormTest::testDataFormatAgnosticVar
sonnet/src/axis_norm_test.py::LayerNormTest::testInstanceNormCorrectAxis
sonnet/src/axis_norm_test.py::LayerNormTest::testInstanceNormCorrectNCW
sonnet/src/axis_norm_test.py::LayerNormTest::testInvalidAxisListString
sonnet/src/axis_norm_test.py::LayerNormTest::testInvalidAxisString
sonnet/src/axis_norm_test.py::LayerNormTest::testInvalidDataFormat0
sonnet/src/axis_norm_test.py::LayerNormTest::testInvalidDataFormat1
sonnet/src/axis_norm_test.py::LayerNormTest::testInvalidDataFormat2
sonnet/src/axis_norm_test.py::LayerNormTest::testNoOffsetBetaInitProvided
sonnet/src/axis_norm_test.py::LayerNormTest::testNoScaleAndInitProvided
sonnet/src/axis_norm_test.py::LayerNormTest::testRankChanges
sonnet/src/axis_norm_test.py::LayerNormTest::testShapeAgnostic
sonnet/src/axis_norm_test.py::LayerNormTest::testSimpleCase
sonnet/src/axis_norm_test.py::LayerNormTest::testSimpleCaseNCHWTensor
sonnet/src/axis_norm_test.py::LayerNormTest::testSimpleCaseNCHWVar
sonnet/src/axis_norm_test.py::LayerNormTest::testSimpleCaseTensor
sonnet/src/axis_norm_test.py::LayerNormTest::testSimpleCaseVar
sonnet/src/axis_norm_test.py::LayerNormTest::testSliceAxis
sonnet/src/axis_norm_test.py::LayerNormTest::testValidDataFormatChannelsFirst0
sonnet/src/axis_norm_test.py::LayerNormTest::testValidDataFormatChannelsFirst1
sonnet/src/axis_norm_test.py::LayerNormTest::testValidDataFormatChannelsFirst2
sonnet/src/axis_norm_test.py::LayerNormTest::testValidDataFormatChannelsLast0
sonnet/src/axis_norm_test.py::LayerNormTest::testValidDataFormatChannelsLast1
sonnet/src/axis_norm_test.py::LayerNormTest::testValidDataFormatChannelsLast2
sonnet/src/axis_norm_test.py::LayerNormTest::testWorksWithFunction
sonnet/src/axis_norm_test.py::LayerNormTest::test_session
sonnet/src/base_test.py::BaseTest::testControlFlow
sonnet/src/base_test.py::BaseTest::testWrappedMethod
sonnet/src/base_test.py::BaseTest::test_basic
sonnet/src/base_test.py::BaseTest::test_session
sonnet/src/base_test.py::TestModuleNaming::test_construct_in_scope
sonnet/src/base_test.py::TestModuleNaming::test_ctor_error_closes_name_scope
sonnet/src/base_test.py::TestModuleNaming::test_ctor_error_handles_ctor_not_opening_name_scope
sonnet/src/base_test.py::TestModuleNaming::test_ctor_no_name_scope
sonnet/src/base_test.py::TestModuleNaming::test_ctor_no_name_scope_no_super
sonnet/src/base_test.py::TestModuleNaming::test_does_not_evaluate_property_methods
sonnet/src/base_test.py::TestModuleNaming::test_enters_name_scope_in_call
sonnet/src/base_test.py::TestModuleNaming::test_enters_name_scope_in_other_method
sonnet/src/base_test.py::TestModuleNaming::test_forward_method_closes_name_scope
sonnet/src/base_test.py::TestModuleNaming::test_get_attr_doesnt_enter_name_scope
sonnet/src/base_test.py::TestModuleNaming::test_get_attribute_doesnt_enter_name_scope
sonnet/src/base_test.py::TestModuleNaming::test_invalid_name
sonnet/src/base_test.py::TestModuleNaming::test_module_numbering_in_graph
sonnet/src/base_test.py::TestModuleNaming::test_modules_not_numbered_in_eager
sonnet/src/base_test.py::TestModuleNaming::test_overridden_name_scope
sonnet/src/base_test.py::TestModuleNaming::test_patched_callable
sonnet/src/base_test.py::TestModuleNaming::test_property
sonnet/src/base_test.py::TestModuleNaming::test_property_no_name_scope
sonnet/src/base_test.py::TestModuleNaming::test_session
sonnet/src/base_test.py::TestModuleNaming::test_single_name
sonnet/src/base_test.py::TestModuleNaming::test_subclassed_module
sonnet/src/base_test.py::TestModuleNaming::test_submodule_created_late
sonnet/src/base_test.py::VariableNamingTest::test_session
sonnet/src/base_test.py::VariableNamingTest::test_variable_names
sonnet/src/base_test.py::AutoReprTest::test_args_are_repred
sonnet/src/base_test.py::AutoReprTest::test_defaults_ignored
sonnet/src/base_test.py::AutoReprTest::test_does_not_fail_with_hostile_input
sonnet/src/base_test.py::AutoReprTest::test_long_repr_multi_line
sonnet/src/base_test.py::AutoReprTest::test_order_matches_argspec
sonnet/src/base_test.py::AutoReprTest::test_repr_non_bool_equality
sonnet/src/base_test.py::AutoReprTest::test_repr_wildcard
sonnet/src/base_test.py::AutoReprTest::test_session
sonnet/src/base_test.py::ForwardMethodsTest::testEntersNameScope_call
sonnet/src/base_test.py::ForwardMethodsTest::testEntersNameScope_concreteFunction
sonnet/src/base_test.py::ForwardMethodsTest::testFunctionType
sonnet/src/base_test.py::ForwardMethodsTest::test_session
sonnet/src/base_test.py::AbcTest::testAbstract
sonnet/src/base_test.py::AbcTest::testCallMethodsOnParent
sonnet/src/base_test.py::AbcTest::testConcrete
sonnet/src/base_test.py::AbcTest::test_session
sonnet/src/base_test.py::CustomGradientTest::test_custom_gradient
sonnet/src/base_test.py::CustomGradientTest::test_session
sonnet/src/base_test.py::CommonErrorsTest::test_allow_empty_variables_class0
sonnet/src/base_test.py::CommonErrorsTest::test_allow_empty_variables_class1
sonnet/src/base_test.py::CommonErrorsTest::test_allow_empty_variables_instance0
sonnet/src/base_test.py::CommonErrorsTest::test_allow_empty_variables_instance1
sonnet/src/base_test.py::CommonErrorsTest::test_annotated_method_is_allowed
sonnet/src/base_test.py::CommonErrorsTest::test_calls_method_before_super
sonnet/src/base_test.py::CommonErrorsTest::test_not_calling_super_constructor
sonnet/src/base_test.py::CommonErrorsTest::test_requests_variables_before_they_exist0
sonnet/src/base_test.py::CommonErrorsTest::test_requests_variables_before_they_exist1
sonnet/src/base_test.py::CommonErrorsTest::test_session
sonnet/src/base_test.py::CustomMetaclassTest::testSupportsCustomMetaclass
sonnet/src/base_test.py::CustomMetaclassTest::test_session
sonnet/src/batch_apply_test.py::BatchApplyTest::test_kwargs
sonnet/src/batch_apply_test.py::BatchApplyTest::test_no_output
sonnet/src/batch_apply_test.py::BatchApplyTest::test_session
sonnet/src/batch_apply_test.py::BatchApplyTest::test_simple
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape0
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape1
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape2
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape3
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape_has_static_info_in_graph0
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape_has_static_info_in_graph1
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape_has_static_info_in_graph2
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_dynamic_shape_has_static_info_in_graph3
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_session
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_static_shape0
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_static_shape1
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_static_shape2
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_static_shape3
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_x_not_tensor0
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_x_not_tensor1
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_x_not_tensor2
sonnet/src/batch_apply_test.py::MergeLeadingDimsTest::test_x_not_tensor3
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape0
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape1
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape2
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape3
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape_has_static_info_in_graph0
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape_has_static_info_in_graph1
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape_has_static_info_in_graph2
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_dynamic_shape_has_static_info_in_graph3
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_session
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_static_shape0
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_static_shape1
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_static_shape2
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_static_shape3
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_x_not_tensor0
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_x_not_tensor1
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_x_not_tensor2
sonnet/src/batch_apply_test.py::SplitLeadingDimTest::test_x_not_tensor3
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testInvalidDataFormat0
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testInvalidDataFormat1
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testInvalidDataFormat2
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testIsTrainingFalseFirstCall
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testNoOffsetBetaInitProvided
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testNoScaleAndInitProvided
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testNoScaleAndOffset
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testSimpleTraining
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testSimpleTraining3D
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testSimpleTraining3DNCDHW
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testSimpleTrainingNCHW
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testUsingTestStats
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testValidDataFormatChannelsFirst0
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testValidDataFormatChannelsFirst1
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testValidDataFormatChannelsFirst2
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testValidDataFormatChannelsLast0
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testValidDataFormatChannelsLast1
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testValidDataFormatChannelsLast2
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testWithTfFunction0
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testWithTfFunction1
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testWithTfFunctionTfArgs0
sonnet/src/batch_norm_test.py::BaseBatchNormTest::testWithTfFunctionTfArgs1
sonnet/src/batch_norm_test.py::BaseBatchNormTest::test_session
sonnet/src/batch_norm_test.py::BatchNormTest::testSimple
sonnet/src/batch_norm_test.py::BatchNormTest::test_session
sonnet/src/bias_test.py::BiasTest::test_b_init_custom
sonnet/src/bias_test.py::BiasTest::test_b_init_defaults_to_zeros
sonnet/src/bias_test.py::BiasTest::test_bias_dims_custom
sonnet/src/bias_test.py::BiasTest::test_bias_dims_invalid
sonnet/src/bias_test.py::BiasTest::test_bias_dims_negative_out_of_order
sonnet/src/bias_test.py::BiasTest::test_bias_dims_scalar
sonnet/src/bias_test.py::BiasTest::test_multiplier
sonnet/src/bias_test.py::BiasTest::test_name
sonnet/src/bias_test.py::BiasTest::test_output_shape
sonnet/src/bias_test.py::BiasTest::test_output_size_valid
sonnet/src/bias_test.py::BiasTest::test_session
sonnet/src/build_test.py::BuildTest::test_call_with_shape_lke_object
sonnet/src/build_test.py::BuildTest::test_does_not_trigger_sideeffects
sonnet/src/build_test.py::BuildTest::test_output_spec
sonnet/src/build_test.py::BuildTest::test_session
sonnet/src/conv_test.py::ConvTest::testDefaultInitializers0
sonnet/src/conv_test.py::ConvTest::testDefaultInitializers1
sonnet/src/conv_test.py::ConvTest::testFunctionSamePaddingNoBias
sonnet/src/conv_test.py::ConvTest::testFunctionSamePaddingUseBias
sonnet/src/conv_test.py::ConvTest::testFunctionValidPaddingNoBias
sonnet/src/conv_test.py::ConvTest::testFunctionValidPaddingUseBias
sonnet/src/conv_test.py::ConvTest::testFunctionsamePaddingNoBias
sonnet/src/conv_test.py::ConvTest::testFunctionsamePaddingUseBias
sonnet/src/conv_test.py::ConvTest::testFunctionvalidPaddingNoBias
sonnet/src/conv_test.py::ConvTest::testFunctionvalidPaddingUseBias
sonnet/src/conv_test.py::ConvTest::testIncorrectN0
sonnet/src/conv_test.py::ConvTest::testIncorrectN1
sonnet/src/conv_test.py::ConvTest::testIncorrectRankInput
sonnet/src/conv_test.py::ConvTest::testInitializerKeysInvalidWithoutBias
sonnet/src/conv_test.py::ConvTest::testPaddingFunctionReached
sonnet/src/conv_test.py::ConvTest::testUnknownBatchSizeNCHW
sonnet/src/conv_test.py::ConvTest::testUnknownBatchSizeNHWC
sonnet/src/conv_test.py::ConvTest::testUnknownChannels0
sonnet/src/conv_test.py::ConvTest::testUnknownChannels1
sonnet/src/conv_test.py::ConvTest::testUnknownSpatialDims
sonnet/src/conv_test.py::ConvTest::test_session
sonnet/src/conv_test.py::Conv2DTest::testComputationPaddingSame0
sonnet/src/conv_test.py::Conv2DTest::testComputationPaddingSame1
sonnet/src/conv_test.py::Conv2DTest::testComputationPaddingValid0
sonnet/src/conv_test.py::Conv2DTest::testComputationPaddingValid1
sonnet/src/conv_test.py::Conv2DTest::test_session
sonnet/src/conv_test.py::Conv1DTest::testComputationPaddingSame0
sonnet/src/conv_test.py::Conv1DTest::testComputationPaddingSame1
sonnet/src/conv_test.py::Conv1DTest::testComputationPaddingValid0
sonnet/src/conv_test.py::Conv1DTest::testComputationPaddingValid1
sonnet/src/conv_test.py::Conv1DTest::test_session
sonnet/src/conv_test.py::Conv3DTest::testComputationPaddingSame0
sonnet/src/conv_test.py::Conv3DTest::testComputationPaddingSame1
sonnet/src/conv_test.py::Conv3DTest::testComputationPaddingValid0
sonnet/src/conv_test.py::Conv3DTest::testComputationPaddingValid1
sonnet/src/conv_test.py::Conv3DTest::test_session
sonnet/src/conv_transpose_test.py::ConvTransposeTest::testGraphConv0
sonnet/src/conv_transpose_test.py::ConvTransposeTest::testGraphConv1
sonnet/src/conv_transpose_test.py::ConvTransposeTest::testGraphConv2
sonnet/src/conv_transpose_test.py::ConvTransposeTest::testGraphConv3
sonnet/src/conv_transpose_test.py::ConvTransposeTest::testIncorrectN0
sonnet/src/conv_transpose_test.py::ConvTransposeTest::testIncorrectN1
sonnet/src/conv_transpose_test.py::ConvTransposeTest::testIncorrectPadding
sonnet/src/conv_transpose_test.py::ConvTransposeTest::testInitializerVariance0
sonnet/src/conv_transpose_test.py::ConvTransposeTest::testInitializerVariance1
sonnet/src/conv_transpose_test.py::ConvTransposeTest::testInitializerVariance2
sonnet/src/conv_transpose_test.py::ConvTransposeTest::testUnknownBatchSizeNCHW
sonnet/src/conv_transpose_test.py::ConvTransposeTest::testUnknownBatchSizeNHWC
sonnet/src/conv_transpose_test.py::ConvTransposeTest::testUnknownChannels0
sonnet/src/conv_transpose_test.py::ConvTransposeTest::testUnknownChannels1
sonnet/src/conv_transpose_test.py::ConvTransposeTest::test_session
sonnet/src/conv_transpose_test.py::Conv2DTransposeTest::testComputationPaddingSame0
sonnet/src/conv_transpose_test.py::Conv2DTransposeTest::testComputationPaddingSame1
sonnet/src/conv_transpose_test.py::Conv2DTransposeTest::testComputationPaddingValid0
sonnet/src/conv_transpose_test.py::Conv2DTransposeTest::testComputationPaddingValid1
sonnet/src/conv_transpose_test.py::Conv2DTransposeTest::testShapeDilated
sonnet/src/conv_transpose_test.py::Conv2DTransposeTest::test_session
sonnet/src/conv_transpose_test.py::Conv1DTransposeTest::testComputationPaddingSame0
sonnet/src/conv_transpose_test.py::Conv1DTransposeTest::testComputationPaddingSame1
sonnet/src/conv_transpose_test.py::Conv1DTransposeTest::testComputationPaddingValid0
sonnet/src/conv_transpose_test.py::Conv1DTransposeTest::testComputationPaddingValid1
sonnet/src/conv_transpose_test.py::Conv1DTransposeTest::test_session
sonnet/src/conv_transpose_test.py::Conv3DTransposeTest::testComputationPaddingSame0
sonnet/src/conv_transpose_test.py::Conv3DTransposeTest::testComputationPaddingSame1
sonnet/src/conv_transpose_test.py::Conv3DTransposeTest::testComputationPaddingValid0
sonnet/src/conv_transpose_test.py::Conv3DTransposeTest::testComputationPaddingValid1
sonnet/src/conv_transpose_test.py::Conv3DTransposeTest::test_session
sonnet/src/custom_getter_test.py::CustomVariableGetterTest::testDoesNotModifyNonVariables
sonnet/src/custom_getter_test.py::CustomVariableGetterTest::test_session
sonnet/src/custom_getter_test.py::DoctestTest::testDoctest
sonnet/src/custom_getter_test.py::DoctestTest::test_session
sonnet/src/deferred_test.py::DeferredTest::test_alternative_forward
sonnet/src/deferred_test.py::DeferredTest::test_alternative_forward_call_type_error
sonnet/src/deferred_test.py::DeferredTest::test_attr_forwarding_fails_before_construction
sonnet/src/deferred_test.py::DeferredTest::test_delattr
sonnet/src/deferred_test.py::DeferredTest::test_getattr
sonnet/src/deferred_test.py::DeferredTest::test_name_scope
sonnet/src/deferred_test.py::DeferredTest::test_only_computes_target_once
sonnet/src/deferred_test.py::DeferredTest::test_repr
sonnet/src/deferred_test.py::DeferredTest::test_session
sonnet/src/deferred_test.py::DeferredTest::test_setattr
sonnet/src/deferred_test.py::DeferredTest::test_setattr_on_target
sonnet/src/deferred_test.py::DeferredTest::test_str
sonnet/src/deferred_test.py::DeferredTest::test_target
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testComputationSameWithBias
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testComputationSameWithoutBias
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testComputationValidMultiChannelWithBias
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testComputationValidMultiChannelWithoutBias
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testComputationValidWithBias
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testComputationValidWithoutBias
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testDefaultInitializers0
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testDefaultInitializers1
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testFunctionSamePaddingNoBias
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testFunctionSamePaddingUseBias
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testFunctionValidPaddingNoBias
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testFunctionValidPaddingUseBias
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testInitializerKeysInvalidWithoutBias
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testSharingWithBias
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testSharingWithoutBias
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testUnknownBatchSizeNCHW
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testUnknownBatchSizeNHWC
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testUnknownChannels0
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testUnknownChannels1
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::testUnknownSpatialDims
sonnet/src/depthwise_conv_test.py::DepthwiseConvTest::test_session
sonnet/src/dropout_test.py::DropoutTest::test_dropout_is_actually_random
sonnet/src/dropout_test.py::DropoutTest::test_dropout_rate0
sonnet/src/dropout_test.py::DropoutTest::test_dropout_rate1
sonnet/src/dropout_test.py::DropoutTest::test_dropout_rate2
sonnet/src/dropout_test.py::DropoutTest::test_dropout_rate3
sonnet/src/dropout_test.py::DropoutTest::test_dropout_rate4
sonnet/src/dropout_test.py::DropoutTest::test_dropout_rate5
sonnet/src/dropout_test.py::DropoutTest::test_dropout_rate6
sonnet/src/dropout_test.py::DropoutTest::test_dropout_rate7
sonnet/src/dropout_test.py::DropoutTest::test_dropout_rate8
sonnet/src/dropout_test.py::DropoutTest::test_session
sonnet/src/dropout_test.py::DropoutTest::test_sum_close0
sonnet/src/dropout_test.py::DropoutTest::test_sum_close1
sonnet/src/dropout_test.py::DropoutTest::test_sum_close2
sonnet/src/dropout_test.py::DropoutTest::test_sum_close3
sonnet/src/dropout_test.py::DropoutTest::test_sum_close4
sonnet/src/dropout_test.py::DropoutTest::test_sum_close5
sonnet/src/dropout_test.py::DropoutTest::test_sum_close6
sonnet/src/dropout_test.py::DropoutTest::test_sum_close7
sonnet/src/dropout_test.py::DropoutTest::test_sum_close8
sonnet/src/dropout_test.py::DropoutTest::test_with_tf_function_with_booleans0
sonnet/src/dropout_test.py::DropoutTest::test_with_tf_function_with_booleans1
sonnet/src/dropout_test.py::DropoutTest::test_with_tf_function_with_variables0
sonnet/src/dropout_test.py::DropoutTest::test_with_tf_function_with_variables1
sonnet/src/embed_test.py::EmbedTest::test_densify_gradients0
sonnet/src/embed_test.py::EmbedTest::test_densify_gradients1
sonnet/src/embed_test.py::EmbedTest::test_dtype0
sonnet/src/embed_test.py::EmbedTest::test_dtype1
sonnet/src/embed_test.py::EmbedTest::test_embed_dim0
sonnet/src/embed_test.py::EmbedTest::test_embed_dim1
sonnet/src/embed_test.py::EmbedTest::test_embed_dim2
sonnet/src/embed_test.py::EmbedTest::test_existing_vocab0
sonnet/src/embed_test.py::EmbedTest::test_existing_vocab1
sonnet/src/embed_test.py::EmbedTest::test_existing_vocab2
sonnet/src/embed_test.py::EmbedTest::test_initializer
sonnet/src/embed_test.py::EmbedTest::test_name
sonnet/src/embed_test.py::EmbedTest::test_pinned_to_cpu
sonnet/src/embed_test.py::EmbedTest::test_session
sonnet/src/embed_test.py::EmbedTest::test_trainable0
sonnet/src/embed_test.py::EmbedTest::test_trainable1
sonnet/src/embed_test.py::EmbedTest::test_vocab_size0
sonnet/src/embed_test.py::EmbedTest::test_vocab_size1
sonnet/src/embed_test.py::EmbedTest::test_vocab_size2
sonnet/src/group_norm_test.py::GroupNormTest::test3DDataFormatAgnostic
sonnet/src/group_norm_test.py::GroupNormTest::test5DDataFormatAgnostic
sonnet/src/group_norm_test.py::GroupNormTest::testBatchSizeAgnostic
sonnet/src/group_norm_test.py::GroupNormTest::testCreateOffsetAndOffsetProvided
sonnet/src/group_norm_test.py::GroupNormTest::testCreateScaleAndScaleProvided
sonnet/src/group_norm_test.py::GroupNormTest::testDataFormatAgnosticTensor
sonnet/src/group_norm_test.py::GroupNormTest::testDataFormatAgnosticVar
sonnet/src/group_norm_test.py::GroupNormTest::testIncompatibleGroupsAndTensorBigger
sonnet/src/group_norm_test.py::GroupNormTest::testIncompatibleGroupsAndTensorSmall
sonnet/src/group_norm_test.py::GroupNormTest::testInvalidAxisListString
sonnet/src/group_norm_test.py::GroupNormTest::testInvalidAxisString
sonnet/src/group_norm_test.py::GroupNormTest::testInvalidDataFormat0
sonnet/src/group_norm_test.py::GroupNormTest::testInvalidDataFormat1
sonnet/src/group_norm_test.py::GroupNormTest::testInvalidDataFormat2
sonnet/src/group_norm_test.py::GroupNormTest::testNoOffsetBetaInitProvided
sonnet/src/group_norm_test.py::GroupNormTest::testNoScaleAndInitProvided
sonnet/src/group_norm_test.py::GroupNormTest::testRankChanges
sonnet/src/group_norm_test.py::GroupNormTest::testSimpleCase
sonnet/src/group_norm_test.py::GroupNormTest::testSimpleCaseNCHWTensor
sonnet/src/group_norm_test.py::GroupNormTest::testSimpleCaseNCHWVar
sonnet/src/group_norm_test.py::GroupNormTest::testSimpleCaseTensor
sonnet/src/group_norm_test.py::GroupNormTest::testSimpleCaseVar
sonnet/src/group_norm_test.py::GroupNormTest::testSliceAxis
sonnet/src/group_norm_test.py::GroupNormTest::testValidDataFormatChannelsFirst0
sonnet/src/group_norm_test.py::GroupNormTest::testValidDataFormatChannelsFirst1
sonnet/src/group_norm_test.py::GroupNormTest::testValidDataFormatChannelsFirst2
sonnet/src/group_norm_test.py::GroupNormTest::testValidDataFormatChannelsLast0
sonnet/src/group_norm_test.py::GroupNormTest::testValidDataFormatChannelsLast1
sonnet/src/group_norm_test.py::GroupNormTest::testValidDataFormatChannelsLast2
sonnet/src/group_norm_test.py::GroupNormTest::testWorksWithFunction
sonnet/src/group_norm_test.py::GroupNormTest::test_session
sonnet/src/initializers_test.py::InitializersTest::test_session
sonnet/src/initializers_test.py::ConstantInitializersTest::testBatchAgnostic
sonnet/src/initializers_test.py::ConstantInitializersTest::testConstantInvalidValueList
sonnet/src/initializers_test.py::ConstantInitializersTest::testConstantInvalidValueTensor
sonnet/src/initializers_test.py::ConstantInitializersTest::testConstantInvalidValueTuple
sonnet/src/initializers_test.py::ConstantInitializersTest::testConstantInvalidValueVariable
sonnet/src/initializers_test.py::ConstantInitializersTest::testConstantValidValue0
sonnet/src/initializers_test.py::ConstantInitializersTest::testConstantValidValue1
sonnet/src/initializers_test.py::ConstantInitializersTest::testConstantValidValue2
sonnet/src/initializers_test.py::ConstantInitializersTest::testInvalidDataType0
sonnet/src/initializers_test.py::ConstantInitializersTest::testInvalidDataType1
sonnet/src/initializers_test.py::ConstantInitializersTest::testInvalidDataTypeConstant
sonnet/src/initializers_test.py::ConstantInitializersTest::testOnes0
sonnet/src/initializers_test.py::ConstantInitializersTest::testOnes1
sonnet/src/initializers_test.py::ConstantInitializersTest::testTFFunction
sonnet/src/initializers_test.py::ConstantInitializersTest::testZeros0
sonnet/src/initializers_test.py::ConstantInitializersTest::testZeros1
sonnet/src/initializers_test.py::ConstantInitializersTest::test_session
sonnet/src/initializers_test.py::RandomUniformInitializerTest::testBatchAgnostic
sonnet/src/initializers_test.py::RandomUniformInitializerTest::testDifferentInitializer0
sonnet/src/initializers_test.py::RandomUniformInitializerTest::testDifferentInitializer1
sonnet/src/initializers_test.py::RandomUniformInitializerTest::testInvalidDataType
sonnet/src/initializers_test.py::RandomUniformInitializerTest::testRangeInitializer
sonnet/src/initializers_test.py::RandomUniformInitializerTest::testTFFunction
sonnet/src/initializers_test.py::RandomUniformInitializerTest::test_session
sonnet/src/initializers_test.py::RandomNormalInitializerTest::testBatchAgnostic
sonnet/src/initializers_test.py::RandomNormalInitializerTest::testDifferentInitializer
sonnet/src/initializers_test.py::RandomNormalInitializerTest::testInvalidDataType0
sonnet/src/initializers_test.py::RandomNormalInitializerTest::testInvalidDataType1
sonnet/src/initializers_test.py::RandomNormalInitializerTest::testRangeInitializer
sonnet/src/initializers_test.py::RandomNormalInitializerTest::testTFFunction
sonnet/src/initializers_test.py::RandomNormalInitializerTest::test_session
sonnet/src/initializers_test.py::TruncatedNormalInitializerTest::testBatchAgnostic
sonnet/src/initializers_test.py::TruncatedNormalInitializerTest::testDifferentInitializer
sonnet/src/initializers_test.py::TruncatedNormalInitializerTest::testInvalidDataType0
sonnet/src/initializers_test.py::TruncatedNormalInitializerTest::testInvalidDataType1
sonnet/src/initializers_test.py::TruncatedNormalInitializerTest::testRangeInitializer
sonnet/src/initializers_test.py::TruncatedNormalInitializerTest::testTFFunction
sonnet/src/initializers_test.py::TruncatedNormalInitializerTest::test_session
sonnet/src/initializers_test.py::IdentityInitializerTest::testBatchAgnostic
sonnet/src/initializers_test.py::IdentityInitializerTest::testInvalidDataType
sonnet/src/initializers_test.py::IdentityInitializerTest::testInvalidShape0
sonnet/src/initializers_test.py::IdentityInitializerTest::testInvalidShape1
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange0
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange1
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange10
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange11
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange12
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange13
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange14
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange15
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange2
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange3
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange4
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange5
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange6
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange7
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange8
sonnet/src/initializers_test.py::IdentityInitializerTest::testRange9
sonnet/src/initializers_test.py::IdentityInitializerTest::testTFFunction
sonnet/src/initializers_test.py::IdentityInitializerTest::testTFFunction4D
sonnet/src/initializers_test.py::IdentityInitializerTest::test_session
sonnet/src/initializers_test.py::OrthogonalInitializerTest::testBatchAgnostic
sonnet/src/initializers_test.py::OrthogonalInitializerTest::testDuplicatedInitializer
sonnet/src/initializers_test.py::OrthogonalInitializerTest::testInvalidDataType0
sonnet/src/initializers_test.py::OrthogonalInitializerTest::testInvalidDataType1
sonnet/src/initializers_test.py::OrthogonalInitializerTest::testInvalidShape
sonnet/src/initializers_test.py::OrthogonalInitializerTest::testRangeInitializer
sonnet/src/initializers_test.py::OrthogonalInitializerTest::testShapesValues3DRectangle
sonnet/src/initializers_test.py::OrthogonalInitializerTest::testShapesValues3DSquare
sonnet/src/initializers_test.py::OrthogonalInitializerTest::testShapesValuesSquare
sonnet/src/initializers_test.py::OrthogonalInitializerTest::testShapesValuesTallRectangle
sonnet/src/initializers_test.py::OrthogonalInitializerTest::testShapesValuesWideRectangle
sonnet/src/initializers_test.py::OrthogonalInitializerTest::testTFFunction
sonnet/src/initializers_test.py::OrthogonalInitializerTest::testTFFunctionSimple
sonnet/src/initializers_test.py::OrthogonalInitializerTest::test_session
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic0
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic1
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic2
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic3
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic4
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic5
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic6
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic7
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testBatchAgnostic8
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testCheckInitalizersEmpty
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testCheckInitalizersInvalid
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testCheckInitalizersValidList
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testCheckInitalizersValidSet
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testCheckInitalizersValidTuple
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testCheckInitializersInvalidType
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testGlorotUniform
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testHeNormal
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testHeUniform
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testInvalidDataType0
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testInvalidDataType1
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testLecunNormal
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testLecunUniform
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testMixedShape0
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testMixedShape1
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testMixedShape2
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testMixedShape3
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testMixedShape4
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testMixedShape5
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testMixedShape6
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testMixedShape7
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testMixedShape8
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testNormalDistribution
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testTruncatedNormalDistribution
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testUniformDistribution
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction0
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction1
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction2
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction3
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction4
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction5
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction6
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction7
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::testWithTFFunction8
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::test_GlorotNormal
sonnet/src/initializers_test.py::VarianceScalingInitializerTest::test_session
sonnet/src/leaky_clip_by_value_test.py::LeakyClipByValueTest::test_leaky_clip_by_value_backward0
sonnet/src/leaky_clip_by_value_test.py::LeakyClipByValueTest::test_leaky_clip_by_value_backward1
sonnet/src/leaky_clip_by_value_test.py::LeakyClipByValueTest::test_leaky_clip_by_value_backward2
sonnet/src/leaky_clip_by_value_test.py::LeakyClipByValueTest::test_leaky_clip_by_value_backward3
sonnet/src/leaky_clip_by_value_test.py::LeakyClipByValueTest::test_leaky_clip_by_value_backward4
sonnet/src/leaky_clip_by_value_test.py::LeakyClipByValueTest::test_leaky_clip_by_value_forward
sonnet/src/leaky_clip_by_value_test.py::LeakyClipByValueTest::test_session
sonnet/src/linear_test.py::LinearTest::testBiasZeroInitialized
sonnet/src/linear_test.py::LinearTest::testCall
sonnet/src/linear_test.py::LinearTest::testFunction0
sonnet/src/linear_test.py::LinearTest::testFunction1
sonnet/src/linear_test.py::LinearTest::testIncorrectDims1D
sonnet/src/linear_test.py::LinearTest::testIncorrectDims3D
sonnet/src/linear_test.py::LinearTest::testInitB
sonnet/src/linear_test.py::LinearTest::testInitW
sonnet/src/linear_test.py::LinearTest::testInitializerKeysInvalidWithoutBias
sonnet/src/linear_test.py::LinearTest::testInputSize
sonnet/src/linear_test.py::LinearTest::testOutputSize
sonnet/src/linear_test.py::LinearTest::testParameterDtype0
sonnet/src/linear_test.py::LinearTest::testParameterDtype1
sonnet/src/linear_test.py::LinearTest::testParameterDtype2
sonnet/src/linear_test.py::LinearTest::testParameterShape
sonnet/src/linear_test.py::LinearTest::testParametersCreatedOnce
sonnet/src/linear_test.py::LinearTest::testUnknownBatchSize
sonnet/src/linear_test.py::LinearTest::testUnknownInputSize
sonnet/src/linear_test.py::LinearTest::test_session
sonnet/src/metrics_test.py::SumTest::testInitialize
sonnet/src/metrics_test.py::SumTest::testReset
sonnet/src/metrics_test.py::SumTest::testSimple
sonnet/src/metrics_test.py::SumTest::test_session
sonnet/src/metrics_test.py::MeanTest::testInitialize
sonnet/src/metrics_test.py::MeanTest::testReset
sonnet/src/metrics_test.py::MeanTest::testSimple
sonnet/src/metrics_test.py::MeanTest::test_session
sonnet/src/mixed_precision_test.py::DummyVar::test_session
sonnet/src/mixed_precision_test.py::DummyInput::test_session
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_bfloat16_input_float16_mode_eligible_class0
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_bfloat16_input_float16_mode_eligible_class1
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_float16_input_float32_mode_eligible_class0
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_float16_input_float32_mode_eligible_class1
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_float16_mode_disable_class0
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_float16_mode_disable_class1
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_float16_mode_eligible_multiple_instances_class0
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_float16_mode_eligible_multiple_instances_class1
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_float16_mode_ineligible_multiple_instances_class0
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_float16_mode_ineligible_multiple_instances_class1
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_float16_mode_multiple_instances_different_eligibility_class0
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_float16_mode_multiple_instances_different_eligibility_class1
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_float16_mode_nested_eligible_class0
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_float16_mode_nested_eligible_class1
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_float16_mode_variable_eligible_class0
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_float16_mode_variable_eligible_class1
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_function_create_module_eligible0
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_function_create_module_eligible1
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_function_create_module_ineligible0
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_function_create_module_ineligible1
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_function_create_module_not_decorated0
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_function_create_module_not_decorated1
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_nested_scoping0
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_nested_scoping1
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_scoping_disable0
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_scoping_disable1
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_scoping_option0
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_scoping_option1
sonnet/src/mixed_precision_test.py::MixedPrecisionClassTest::test_session
sonnet/src/mixed_precision_test.py::MixedPrecisionTest::test_dont_cast_non_floats_func
sonnet/src/mixed_precision_test.py::MixedPrecisionTest::test_float16_mode_eligible_func
sonnet/src/mixed_precision_test.py::MixedPrecisionTest::test_float16_mode_enabled_call_function
sonnet/src/mixed_precision_test.py::MixedPrecisionTest::test_float16_mode_ineligible_func
sonnet/src/mixed_precision_test.py::MixedPrecisionTest::test_float16_mode_tensor_eligible_class
sonnet/src/mixed_precision_test.py::MixedPrecisionTest::test_float32_mode_eligible_func
sonnet/src/mixed_precision_test.py::MixedPrecisionTest::test_non_tensor_variable_input_no_cast_func
sonnet/src/mixed_precision_test.py::MixedPrecisionTest::test_session
sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testAlternativeShape2D
sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testAlternativeShape3D
sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testCall
sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testReset
sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testResetVector
sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testResetWithTFFunction0
sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testResetWithTFFunction1
sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testUpdateAndValue
sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testValueEqualsLatestUpdate
sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testWithTFFunction0
sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::testWithTFFunction1
sonnet/src/moving_averages_test.py::ExponentialMovingAverageTest::test_session
sonnet/src/once_test.py::OnceTest::test_adds_property_function
sonnet/src/once_test.py::OnceTest::test_adds_property_lambda
sonnet/src/once_test.py::OnceTest::test_adds_property_method
sonnet/src/once_test.py::OnceTest::test_adds_property_object
sonnet/src/once_test.py::OnceTest::test_adds_property_special_method
sonnet/src/once_test.py::OnceTest::test_always_returns_none
sonnet/src/once_test.py::OnceTest::test_does_not_cache_on_error
sonnet/src/once_test.py::OnceTest::test_inline
sonnet/src/once_test.py::OnceTest::test_method
sonnet/src/once_test.py::OnceTest::test_method_does_not_cache_on_error
sonnet/src/once_test.py::OnceTest::test_pickle_method_already_evaluated
sonnet/src/once_test.py::OnceTest::test_pickle_method_before_evaluation
sonnet/src/once_test.py::OnceTest::test_runs_once
sonnet/src/pad_test.py::PadTest::test_padding_1d
sonnet/src/pad_test.py::PadTest::test_padding_2d
sonnet/src/pad_test.py::PadTest::test_padding_3d
sonnet/src/pad_test.py::PadTest::test_padding_causal
sonnet/src/pad_test.py::PadTest::test_padding_full
sonnet/src/pad_test.py::PadTest::test_padding_incorrect_input0
sonnet/src/pad_test.py::PadTest::test_padding_incorrect_input1
sonnet/src/pad_test.py::PadTest::test_padding_incorrect_input2
sonnet/src/pad_test.py::PadTest::test_padding_incorrect_input3
sonnet/src/pad_test.py::PadTest::test_padding_reverse_causal
sonnet/src/pad_test.py::PadTest::test_padding_same
sonnet/src/pad_test.py::PadTest::test_padding_valid
sonnet/src/pad_test.py::PadTest::test_same_padding0
sonnet/src/pad_test.py::PadTest::test_same_padding1
sonnet/src/pad_test.py::PadTest::test_same_padding2
sonnet/src/pad_test.py::PadTest::test_same_padding3
sonnet/src/pad_test.py::PadTest::test_same_padding4
sonnet/src/pad_test.py::PadTest::test_same_padding5
sonnet/src/pad_test.py::PadTest::test_same_padding6
sonnet/src/pad_test.py::PadTest::test_same_padding7
sonnet/src/pad_test.py::PadTest::test_session
sonnet/src/pad_test.py::PadTest::test_valid_padding0
sonnet/src/pad_test.py::PadTest::test_valid_padding1
sonnet/src/pad_test.py::PadTest::test_valid_padding2
sonnet/src/pad_test.py::PadTest::test_valid_padding3
sonnet/src/pad_test.py::PadTest::test_valid_padding4
sonnet/src/pad_test.py::PadTest::test_valid_padding5
sonnet/src/pad_test.py::PadTest::test_valid_padding6
sonnet/src/pad_test.py::PadTest::test_valid_padding7
sonnet/src/parallel_linear_test.py::ParallelLinearTest::test_behaves_same_as_stacked_linears
sonnet/src/parallel_linear_test.py::ParallelLinearTest::test_output_size_correct
sonnet/src/parallel_linear_test.py::ParallelLinearTest::test_session
sonnet/src/recurrent_test.py::VanillaRNNTest::testComputationAgainstNumPy0
sonnet/src/recurrent_test.py::VanillaRNNTest::testComputationAgainstNumPy1
sonnet/src/recurrent_test.py::VanillaRNNTest::testDtypeMismatch
sonnet/src/recurrent_test.py::VanillaRNNTest::testInitialization
sonnet/src/recurrent_test.py::VanillaRNNTest::test_session
sonnet/src/recurrent_test.py::DeepRNNTest::testComputationAgainstNumPy0
sonnet/src/recurrent_test.py::DeepRNNTest::testComputationAgainstNumPy1
sonnet/src/recurrent_test.py::DeepRNNTest::testComputationAgainstNumPyWithCallables0
sonnet/src/recurrent_test.py::DeepRNNTest::testComputationAgainstNumPyWithCallables1
sonnet/src/recurrent_test.py::DeepRNNTest::testInitialState
sonnet/src/recurrent_test.py::DeepRNNTest::testWithConnectionsValidation
sonnet/src/recurrent_test.py::DeepRNNTest::testWithSkipConnectionsOutputs0
sonnet/src/recurrent_test.py::DeepRNNTest::testWithSkipConnectionsOutputs1
sonnet/src/recurrent_test.py::DeepRNNTest::test_session
sonnet/src/recurrent_test.py::LSTMTest::testComputationAgainstNumPy0
sonnet/src/recurrent_test.py::LSTMTest::testComputationAgainstNumPy1
sonnet/src/recurrent_test.py::LSTMTest::testComputationAgainstNumPy2
sonnet/src/recurrent_test.py::LSTMTest::testComputationAgainstNumPy3
sonnet/src/recurrent_test.py::LSTMTest::testComputationAgainstNumPy4
sonnet/src/recurrent_test.py::LSTMTest::testComputationAgainstNumPy5
sonnet/src/recurrent_test.py::LSTMTest::testComputationAgainstNumPy6
sonnet/src/recurrent_test.py::LSTMTest::testComputationAgainstNumPy7
sonnet/src/recurrent_test.py::LSTMTest::testDtypeMismatch
sonnet/src/recurrent_test.py::LSTMTest::testInitialization
sonnet/src/recurrent_test.py::LSTMTest::testRecurrentDropout0
sonnet/src/recurrent_test.py::LSTMTest::testRecurrentDropout1
sonnet/src/recurrent_test.py::LSTMTest::testRecurrentDropout2
sonnet/src/recurrent_test.py::LSTMTest::testRecurrentDropoutInvalid
sonnet/src/recurrent_test.py::LSTMTest::test_session
sonnet/src/recurrent_test.py::UnrolledLSTMTest::testComputationAgainstLSTM0
sonnet/src/recurrent_test.py::UnrolledLSTMTest::testComputationAgainstLSTM1
sonnet/src/recurrent_test.py::UnrolledLSTMTest::testComputationAgainstLSTM2
sonnet/src/recurrent_test.py::UnrolledLSTMTest::testComputationAgainstLSTM3
sonnet/src/recurrent_test.py::UnrolledLSTMTest::testDtypeMismatch
sonnet/src/recurrent_test.py::UnrolledLSTMTest::testInitialization
sonnet/src/recurrent_test.py::UnrolledLSTMTest::testNumStepsPolymorphism0
sonnet/src/recurrent_test.py::UnrolledLSTMTest::testNumStepsPolymorphism1
sonnet/src/recurrent_test.py::UnrolledLSTMTest::test_session
sonnet/src/recurrent_test.py::ConvNDLSTMTest::testComputationAgainstNumPy0
sonnet/src/recurrent_test.py::ConvNDLSTMTest::testComputationAgainstNumPy1
sonnet/src/recurrent_test.py::ConvNDLSTMTest::testComputationAgainstNumPy2
sonnet/src/recurrent_test.py::ConvNDLSTMTest::testComputationAgainstNumPy3
sonnet/src/recurrent_test.py::ConvNDLSTMTest::testComputationAgainstNumPy4
sonnet/src/recurrent_test.py::ConvNDLSTMTest::testComputationAgainstNumPy5
sonnet/src/recurrent_test.py::ConvNDLSTMTest::testDtypeMismatch
sonnet/src/recurrent_test.py::ConvNDLSTMTest::testInitialization
sonnet/src/recurrent_test.py::ConvNDLSTMTest::test_session
sonnet/src/recurrent_test.py::GRUTest::testComputationAgainstNumPy0
sonnet/src/recurrent_test.py::GRUTest::testComputationAgainstNumPy1
sonnet/src/recurrent_test.py::GRUTest::testDtypeMismatch
sonnet/src/recurrent_test.py::GRUTest::testInitialization
sonnet/src/recurrent_test.py::GRUTest::test_session
sonnet/src/recurrent_test.py::CuDNNGRUTest::testComputationAgainstTF0
sonnet/src/recurrent_test.py::CuDNNGRUTest::testComputationAgainstTF1
sonnet/src/recurrent_test.py::CuDNNGRUTest::testDtypeMismatch
sonnet/src/recurrent_test.py::CuDNNGRUTest::testInitialization
sonnet/src/recurrent_test.py::CuDNNGRUTest::test_session
sonnet/src/recurrent_test.py::TrainableStateTest::testForCore
sonnet/src/recurrent_test.py::TrainableStateTest::testMasked
sonnet/src/recurrent_test.py::TrainableStateTest::testUnmasked0
sonnet/src/recurrent_test.py::TrainableStateTest::testUnmasked1
sonnet/src/recurrent_test.py::TrainableStateTest::testUnmasked2
sonnet/src/recurrent_test.py::TrainableStateTest::test_session
sonnet/src/recurrent_test.py::UnrollTest::testEmptyOutputs0
sonnet/src/recurrent_test.py::UnrollTest::testEmptyOutputs1
sonnet/src/recurrent_test.py::UnrollTest::testEmptyOutputs2
sonnet/src/recurrent_test.py::UnrollTest::testEmptyOutputs3
sonnet/src/recurrent_test.py::UnrollTest::testFlat0
sonnet/src/recurrent_test.py::UnrollTest::testFlat1
sonnet/src/recurrent_test.py::UnrollTest::testFlat2
sonnet/src/recurrent_test.py::UnrollTest::testFlat3
sonnet/src/recurrent_test.py::UnrollTest::testInconsistentSteps0
sonnet/src/recurrent_test.py::UnrollTest::testInconsistentSteps1
sonnet/src/recurrent_test.py::UnrollTest::testInconsistentSteps2
sonnet/src/recurrent_test.py::UnrollTest::testInconsistentSteps3
sonnet/src/recurrent_test.py::UnrollTest::testNestedInputs0
sonnet/src/recurrent_test.py::UnrollTest::testNestedInputs1
sonnet/src/recurrent_test.py::UnrollTest::testNestedInputs2
sonnet/src/recurrent_test.py::UnrollTest::testNestedInputs3
sonnet/src/recurrent_test.py::UnrollTest::testNestedOutputs0
sonnet/src/recurrent_test.py::UnrollTest::testNestedOutputs1
sonnet/src/recurrent_test.py::UnrollTest::testNestedOutputs2
sonnet/src/recurrent_test.py::UnrollTest::testNestedOutputs3
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthAllEmpty0
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthAllEmpty1
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthAllEmpty2
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthAllEmpty3
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthAllFull0
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthAllFull1
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthAllFull2
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthAllFull3
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthOneZeroLength0
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthOneZeroLength1
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthOneZeroLength2
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthOneZeroLength3
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthRange0
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthRange1
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthRange2
sonnet/src/recurrent_test.py::UnrollTest::testVariableLengthRange3
sonnet/src/recurrent_test.py::UnrollTest::testZeroSteps0
sonnet/src/recurrent_test.py::UnrollTest::testZeroSteps1
sonnet/src/recurrent_test.py::UnrollTest::testZeroSteps2
sonnet/src/recurrent_test.py::UnrollTest::testZeroSteps3
sonnet/src/recurrent_test.py::UnrollTest::test_session
sonnet/src/recurrent_test.py::UnknownStepsUnrollTest::testDynamicUnroll
sonnet/src/recurrent_test.py::UnknownStepsUnrollTest::testDynamicUnrollInconsistentSteps
sonnet/src/recurrent_test.py::UnknownStepsUnrollTest::testStaticUnroll
sonnet/src/recurrent_test.py::UnknownStepsUnrollTest::test_session
sonnet/src/regularizers_test.py::L1Test::testAgainstNumPy
sonnet/src/regularizers_test.py::L1Test::testEmpty
sonnet/src/regularizers_test.py::L1Test::testNegativeScale
sonnet/src/regularizers_test.py::L1Test::test_session
sonnet/src/regularizers_test.py::L2Test::testAgainstNumPy
sonnet/src/regularizers_test.py::L2Test::testEmpty
sonnet/src/regularizers_test.py::L2Test::testNegativeScale
sonnet/src/regularizers_test.py::L2Test::test_session
sonnet/src/regularizers_test.py::OffDiagonalOrthogonalTest::testAgainstNumPy
sonnet/src/regularizers_test.py::OffDiagonalOrthogonalTest::testEmpty
sonnet/src/regularizers_test.py::OffDiagonalOrthogonalTest::testNegativeScale
sonnet/src/regularizers_test.py::OffDiagonalOrthogonalTest::test_session
sonnet/src/reshape_test.py::ReshapeTest::testAddDimensions
sonnet/src/reshape_test.py::ReshapeTest::testBuildDimError
sonnet/src/reshape_test.py::ReshapeTest::testFlatten
sonnet/src/reshape_test.py::ReshapeTest::testIncompatibleShape
sonnet/src/reshape_test.py::ReshapeTest::testInferShape
sonnet/src/reshape_test.py::ReshapeTest::testInvalidPreserveDimsError
sonnet/src/reshape_test.py::ReshapeTest::testInvalid_multipleWildcard
sonnet/src/reshape_test.py::ReshapeTest::testInvalid_negativeSize
sonnet/src/reshape_test.py::ReshapeTest::testInvalid_type
sonnet/src/reshape_test.py::ReshapeTest::testPreservePreserve1
sonnet/src/reshape_test.py::ReshapeTest::testPreservePreserve24
sonnet/src/reshape_test.py::ReshapeTest::testPreservePreserve5?
sonnet/src/reshape_test.py::ReshapeTest::testPreservePreserve?
sonnet/src/reshape_test.py::ReshapeTest::testPreservePreserve?5
sonnet/src/reshape_test.py::ReshapeTest::testPreservePreserve??
sonnet/src/reshape_test.py::ReshapeTest::testReshape0
sonnet/src/reshape_test.py::ReshapeTest::testReshape1
sonnet/src/reshape_test.py::ReshapeTest::testReshape2
sonnet/src/reshape_test.py::ReshapeTest::testReshape3
sonnet/src/reshape_test.py::ReshapeTest::testReverse
sonnet/src/reshape_test.py::ReshapeTest::testReverse_name
sonnet/src/reshape_test.py::ReshapeTest::testRunSession1
sonnet/src/reshape_test.py::ReshapeTest::testRunSession2
sonnet/src/reshape_test.py::ReshapeTest::testRunSession3
sonnet/src/reshape_test.py::ReshapeTest::testRunSession4
sonnet/src/reshape_test.py::ReshapeTest::testRunSession5
sonnet/src/reshape_test.py::ReshapeTest::testRunSession6
sonnet/src/reshape_test.py::ReshapeTest::testRunSession7
sonnet/src/reshape_test.py::ReshapeTest::testRunSession8
sonnet/src/reshape_test.py::ReshapeTest::testRunSession9
sonnet/src/reshape_test.py::ReshapeTest::testUnknownBatchSize
sonnet/src/reshape_test.py::ReshapeTest::test_session
sonnet/src/reshape_test.py::FlattenTest::testBuildDimError
sonnet/src/reshape_test.py::FlattenTest::testFlatten0
sonnet/src/reshape_test.py::FlattenTest::testFlatten1
sonnet/src/reshape_test.py::FlattenTest::testFlattenWithZeroDim
sonnet/src/reshape_test.py::FlattenTest::testFlatten_unknownBatchSize
sonnet/src/reshape_test.py::FlattenTest::testFlatten_unknownNonBatchSize
sonnet/src/reshape_test.py::FlattenTest::testInvalidFlattenFromError
sonnet/src/reshape_test.py::FlattenTest::testPreserveDimsError0
sonnet/src/reshape_test.py::FlattenTest::testPreserveDimsError1
sonnet/src/reshape_test.py::FlattenTest::testPreserveDimsError2
sonnet/src/reshape_test.py::FlattenTest::testPreserveDimsError3
sonnet/src/reshape_test.py::FlattenTest::testPreserveDimsOk0
sonnet/src/reshape_test.py::FlattenTest::testPreserveDimsOk1
sonnet/src/reshape_test.py::FlattenTest::testPreserveDimsOk2
sonnet/src/reshape_test.py::FlattenTest::testPreserveDimsOk3
sonnet/src/reshape_test.py::FlattenTest::testReverse0
sonnet/src/reshape_test.py::FlattenTest::testReverse1
sonnet/src/reshape_test.py::FlattenTest::test_session
sonnet/src/scale_gradient_test.py::ScaleGradientTest::test_scale0
sonnet/src/scale_gradient_test.py::ScaleGradientTest::test_scale1
sonnet/src/scale_gradient_test.py::ScaleGradientTest::test_scale10
sonnet/src/scale_gradient_test.py::ScaleGradientTest::test_scale11
sonnet/src/scale_gradient_test.py::ScaleGradientTest::test_scale2
sonnet/src/scale_gradient_test.py::ScaleGradientTest::test_scale3
sonnet/src/scale_gradient_test.py::ScaleGradientTest::test_scale4
sonnet/src/scale_gradient_test.py::ScaleGradientTest::test_scale5
sonnet/src/scale_gradient_test.py::ScaleGradientTest::test_scale6
sonnet/src/scale_gradient_test.py::ScaleGradientTest::test_scale7
sonnet/src/scale_gradient_test.py::ScaleGradientTest::test_scale8
sonnet/src/scale_gradient_test.py::ScaleGradientTest::test_scale9
sonnet/src/scale_gradient_test.py::ScaleGradientTest::test_session
sonnet/src/sequential_test.py::SequentialTest::test_call
sonnet/src/sequential_test.py::SequentialTest::test_empty0
sonnet/src/sequential_test.py::SequentialTest::test_empty1
sonnet/src/sequential_test.py::SequentialTest::test_empty2
sonnet/src/sequential_test.py::SequentialTest::test_empty3
sonnet/src/sequential_test.py::SequentialTest::test_empty4
sonnet/src/sequential_test.py::SequentialTest::test_empty5
sonnet/src/sequential_test.py::SequentialTest::test_empty_drops_varargs_varkwargs0
sonnet/src/sequential_test.py::SequentialTest::test_empty_drops_varargs_varkwargs1
sonnet/src/sequential_test.py::SequentialTest::test_empty_drops_varargs_varkwargs2
sonnet/src/sequential_test.py::SequentialTest::test_empty_drops_varargs_varkwargs3
sonnet/src/sequential_test.py::SequentialTest::test_empty_drops_varargs_varkwargs4
sonnet/src/sequential_test.py::SequentialTest::test_empty_drops_varargs_varkwargs5
sonnet/src/sequential_test.py::SequentialTest::test_identity_chain0
sonnet/src/sequential_test.py::SequentialTest::test_identity_chain1
sonnet/src/sequential_test.py::SequentialTest::test_identity_chain2
sonnet/src/sequential_test.py::SequentialTest::test_identity_chain3
sonnet/src/sequential_test.py::SequentialTest::test_identity_chain4
sonnet/src/sequential_test.py::SequentialTest::test_identity_chain5
sonnet/src/sequential_test.py::SequentialTest::test_session
sonnet/src/sequential_test.py::SequentialTest::test_varargs_varkwargs_to_call
sonnet/src/test_utils.py::TestCase::test_session
sonnet/src/utils_test.py::ReplicateTest::testIncorrectLength
sonnet/src/utils_test.py::ReplicateTest::testListLengthNCallable
sonnet/src/utils_test.py::ReplicateTest::testListLengthNInt
sonnet/src/utils_test.py::ReplicateTest::testListLengthNString
sonnet/src/utils_test.py::ReplicateTest::testListLengthOneCallable
sonnet/src/utils_test.py::ReplicateTest::testListLengthOneInt
sonnet/src/utils_test.py::ReplicateTest::testListLengthOneString
sonnet/src/utils_test.py::ReplicateTest::testSingleValueCallable
sonnet/src/utils_test.py::ReplicateTest::testSingleValueInt
sonnet/src/utils_test.py::ReplicateTest::testTupleLengthNCallable
sonnet/src/utils_test.py::ReplicateTest::testTupleLengthNInt
sonnet/src/utils_test.py::ReplicateTest::testTupleLengthNString
sonnet/src/utils_test.py::ReplicateTest::test_session
sonnet/src/utils_test.py::DecoratorTest::test_bound_method
sonnet/src/utils_test.py::DecoratorTest::test_callable_object
sonnet/src/utils_test.py::DecoratorTest::test_function
sonnet/src/utils_test.py::DecoratorTest::test_session
sonnet/src/utils_test.py::DecoratorTest::test_unbound_method
sonnet/src/utils_test.py::ChannelIndexTest::test_invalid_strings0
sonnet/src/utils_test.py::ChannelIndexTest::test_invalid_strings1
sonnet/src/utils_test.py::ChannelIndexTest::test_invalid_strings2
sonnet/src/utils_test.py::ChannelIndexTest::test_invalid_strings3
sonnet/src/utils_test.py::ChannelIndexTest::test_invalid_strings4
sonnet/src/utils_test.py::ChannelIndexTest::test_returns_index_channels_first0
sonnet/src/utils_test.py::ChannelIndexTest::test_returns_index_channels_first1
sonnet/src/utils_test.py::ChannelIndexTest::test_returns_index_channels_first2
sonnet/src/utils_test.py::ChannelIndexTest::test_returns_index_channels_first3
sonnet/src/utils_test.py::ChannelIndexTest::test_returns_index_channels_last0
sonnet/src/utils_test.py::ChannelIndexTest::test_returns_index_channels_last1
sonnet/src/utils_test.py::ChannelIndexTest::test_returns_index_channels_last2
sonnet/src/utils_test.py::ChannelIndexTest::test_returns_index_channels_last3
sonnet/src/utils_test.py::ChannelIndexTest::test_returns_index_channels_last4
sonnet/src/utils_test.py::ChannelIndexTest::test_session
sonnet/src/utils_test.py::AssertRankTest::test_invalid_rank0
sonnet/src/utils_test.py::AssertRankTest::test_invalid_rank1
sonnet/src/utils_test.py::AssertRankTest::test_invalid_rank2
sonnet/src/utils_test.py::AssertRankTest::test_invalid_rank3
sonnet/src/utils_test.py::AssertRankTest::test_invalid_rank4
sonnet/src/utils_test.py::AssertRankTest::test_invalid_rank5
sonnet/src/utils_test.py::AssertRankTest::test_invalid_rank6
sonnet/src/utils_test.py::AssertRankTest::test_invalid_rank7
sonnet/src/utils_test.py::AssertRankTest::test_invalid_rank8
sonnet/src/utils_test.py::AssertRankTest::test_invalid_rank9
sonnet/src/utils_test.py::AssertRankTest::test_session
sonnet/src/utils_test.py::AssertRankTest::test_valid_rank_np_ndarray
sonnet/src/utils_test.py::AssertRankTest::test_valid_rank_tf_tensor
sonnet/src/utils_test.py::AssertRankTest::test_valid_rank_tf_tensorspec
sonnet/src/utils_test.py::AssertRankTest::test_valid_rank_tf_variable
sonnet/src/utils_test.py::SmartAutographTest::test_session
sonnet/src/utils_test.py::SmartAutographTest::test_smart_ag
sonnet/src/utils_test.py::VariableLikeTest::test_copies_device0
sonnet/src/utils_test.py::VariableLikeTest::test_copies_device1
sonnet/src/utils_test.py::VariableLikeTest::test_copies_dtype0
sonnet/src/utils_test.py::VariableLikeTest::test_copies_dtype1
sonnet/src/utils_test.py::VariableLikeTest::test_copies_shape0
sonnet/src/utils_test.py::VariableLikeTest::test_copies_shape1
sonnet/src/utils_test.py::VariableLikeTest::test_copies_variable_name
sonnet/src/utils_test.py::VariableLikeTest::test_copies_variable_trainable0
sonnet/src/utils_test.py::VariableLikeTest::test_copies_variable_trainable1
sonnet/src/utils_test.py::VariableLikeTest::test_default_initializer_is_zero
sonnet/src/utils_test.py::VariableLikeTest::test_default_name_for_tensor
sonnet/src/utils_test.py::VariableLikeTest::test_default_trainable_for_tensor
sonnet/src/utils_test.py::VariableLikeTest::test_override_initializer
sonnet/src/utils_test.py::VariableLikeTest::test_override_name0
sonnet/src/utils_test.py::VariableLikeTest::test_override_name1
sonnet/src/utils_test.py::VariableLikeTest::test_override_trainable0
sonnet/src/utils_test.py::VariableLikeTest::test_override_trainable1
sonnet/src/utils_test.py::VariableLikeTest::test_session
sonnet/src/utils_test.py::FormatVariablesTest::test_format_variables
sonnet/src/utils_test.py::FormatVariablesTest::test_log_variables
sonnet/src/utils_test.py::FormatVariablesTest::test_session
sonnet/src/utils_test.py::CompareByIdTest::test_access
sonnet/src/utils_test.py::CompareByIdTest::test_eq
sonnet/src/utils_test.py::CompareByIdTest::test_hash
sonnet/src/utils_test.py::CompareByIdTest::test_session
sonnet/src/conformance/api_test.py::PublicSymbolsTest::test_session
sonnet/src/conformance/api_test.py::PublicSymbolsTest::test_src_not_exported
sonnet/src/conformance/api_test.py::PublicSymbolsTest::test_supports_reload
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_base_batch_norm_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_base_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_batch_norm_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_batch_norm_training_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_bias_3x3x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_cifar10_convnet_2x3_2x2_1x3x3x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv1d_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv1d_lstm_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv1d_transpose_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv2d_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv2d_lstm_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv2d_transpose_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv3d_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv3d_lstm_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_conv3d_transpose_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_cross_replica_batch_norm_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_dropout
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_ema_2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_embed_100_100
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_group_norm_2_1x3x4
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_gru_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_instance_norm_1_1x3_2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_layer_norm_1_1x3_2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_linear_1x1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_linear_nobias_1x1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_lstm_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_lstm_8_projected_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_mean_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_mlp_3x4x5_1x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_mlp_nobias_3x4x5_1x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_resnet50
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_sum_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_trainable_state
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_unrolled_lstm_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_vanilla_rnn_8
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_vqvae
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_vqvae_ema_eval
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_golden_vqvae_ema_train
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_base_batch_norm_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_base_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_batch_norm_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_batch_norm_training_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_bias_3x3x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_cifar10_convnet_2x3_2x2_1x3x3x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv1d_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv1d_lstm_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv1d_transpose_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv2d_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv2d_lstm_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv2d_transpose_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv3d_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv3d_lstm_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_conv3d_transpose_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_cross_replica_batch_norm_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_dropout
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_ema_2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_embed_100_100
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_group_norm_2_1x3x4
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_gru_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_instance_norm_1_1x3_2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_layer_norm_1_1x3_2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_linear_1x1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_linear_nobias_1x1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_lstm_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_lstm_8_projected_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_mean_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_mlp_3x4x5_1x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_mlp_nobias_3x4x5_1x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_resnet50
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_sum_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_trainable_state
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_unrolled_lstm_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_vanilla_rnn_8
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_vqvae
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_vqvae_ema_eval
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_restore_on_create_vqvae_ema_train
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_base_batch_norm_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_base_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_batch_norm_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_batch_norm_training_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_bias_3x3x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_cifar10_convnet_2x3_2x2_1x3x3x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv1d_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv1d_lstm_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv1d_transpose_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv2d_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv2d_lstm_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv2d_transpose_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv3d_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv3d_lstm_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_conv3d_transpose_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_cross_replica_batch_norm_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_dropout
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_ema_2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_embed_100_100
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_group_norm_2_1x3x4
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_gru_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_instance_norm_1_1x3_2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_layer_norm_1_1x3_2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_linear_1x1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_linear_nobias_1x1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_lstm_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_lstm_8_projected_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_mean_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_mlp_3x4x5_1x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_mlp_nobias_3x4x5_1x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_resnet50
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_sum_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_trainable_state
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_unrolled_lstm_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_vanilla_rnn_8
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_vqvae
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_vqvae_ema_eval
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_load_vqvae_ema_train
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_base_batch_norm_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_base_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_batch_norm_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_batch_norm_training_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_bias_3x3x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_cifar10_convnet_2x3_2x2_1x3x3x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv1d_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv1d_lstm_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv1d_transpose_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv2d_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv2d_lstm_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv2d_transpose_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv3d_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv3d_lstm_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_conv3d_transpose_3x3_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_cross_replica_batch_norm_1x2x2x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_dropout
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_ema_2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_embed_100_100
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_group_norm_2_1x3x4
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_gru_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_instance_norm_1_1x3_2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_layer_norm_1_1x3_2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_linear_1x1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_linear_nobias_1x1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_lstm_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_lstm_8_projected_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_mean_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_mlp_3x4x5_1x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_mlp_nobias_3x4x5_1x3
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_resnet50
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_sum_2x2
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_trainable_state
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_unrolled_lstm_1
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_vanilla_rnn_8
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_vqvae
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_vqvae_ema_eval
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_save_then_load_new_instance_vqvae_ema_train
sonnet/src/conformance/checkpoint_test.py::GoldenCheckpointsTest::test_session
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_base_batch_norm_1x2x2x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_base_batch_norm_1x2x2x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_base_batch_norm_scale_offset_1x2x2x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_base_batch_norm_scale_offset_1x2x2x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_batch_norm_1x2x2x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_batch_norm_1x2x2x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_batch_norm_scale_offset_1x2x2x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_batch_norm_scale_offset_1x2x2x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_batch_norm_training_1x2x2x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_batch_norm_training_1x2x2x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_bias_3x3x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_bias_3x3x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_cifar10_convnet_2x3_2x2_1x3x3x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv1d_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv1d_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv1d_lstm_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv1d_lstm_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv1d_transpose_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv1d_transpose_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv2d_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv2d_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv2d_lstm_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv2d_lstm_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv2d_transpose_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv2d_transpose_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv3d_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv3d_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv3d_lstm_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv3d_lstm_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv3d_transpose_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_conv3d_transpose_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_cross_replica_batch_norm_1x2x2x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_cross_replica_batch_norm_1x2x2x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_dropout_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_dropout_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_ema_2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_ema_2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_embed_100_100_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_embed_100_100_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_group_norm_2_1x3x4_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_group_norm_2_1x3x4_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_gru_1_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_gru_1_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_instance_norm_1_1x3_2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_instance_norm_1_1x3_2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_layer_norm_1_1x3_2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_layer_norm_1_1x3_2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_linear_1x1_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_linear_1x1_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_linear_nobias_1x1_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_linear_nobias_1x1_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_lstm_1_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_lstm_1_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_lstm_8_projected_1_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_lstm_8_projected_1_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_mean_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_mean_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_mlp_3x4x5_1x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_mlp_3x4x5_1x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_mlp_nobias_3x4x5_1x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_mlp_nobias_3x4x5_1x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_resnet50_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_resnet50_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_sum_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_sum_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_trainable_state_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_trainable_state_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_unrolled_lstm_1_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_unrolled_lstm_1_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_vanilla_rnn_8_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_vanilla_rnn_8_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_vqvae_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_vqvae_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_vqvae_ema_eval_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_vqvae_ema_eval_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_vqvae_ema_train_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_golden_vqvae_ema_train_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_base_batch_norm_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_base_batch_norm_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_base_batch_norm_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_base_batch_norm_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_base_batch_norm_scale_offset_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_base_batch_norm_scale_offset_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_base_batch_norm_scale_offset_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_base_batch_norm_scale_offset_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_scale_offset_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_scale_offset_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_scale_offset_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_scale_offset_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_training_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_training_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_training_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_batch_norm_training_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_bias_3x3x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_bias_3x3x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_bias_3x3x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_bias_3x3x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_cifar10_convnet_2x3_2x2_1x3x3x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_cifar10_convnet_2x3_2x2_1x3x3x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_lstm_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_lstm_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_lstm_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_lstm_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_transpose_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_transpose_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_transpose_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv1d_transpose_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_lstm_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_lstm_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_lstm_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_lstm_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_transpose_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_transpose_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_transpose_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv2d_transpose_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_lstm_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_lstm_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_lstm_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_lstm_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_transpose_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_transpose_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_transpose_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_conv3d_transpose_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_cross_replica_batch_norm_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_cross_replica_batch_norm_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_cross_replica_batch_norm_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_cross_replica_batch_norm_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_dropout_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_dropout_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_dropout_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_dropout_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_ema_2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_ema_2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_ema_2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_ema_2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_embed_100_100_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_embed_100_100_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_embed_100_100_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_embed_100_100_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_group_norm_2_1x3x4_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_group_norm_2_1x3x4_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_group_norm_2_1x3x4_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_group_norm_2_1x3x4_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_gru_1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_gru_1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_gru_1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_gru_1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_instance_norm_1_1x3_2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_instance_norm_1_1x3_2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_instance_norm_1_1x3_2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_instance_norm_1_1x3_2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_layer_norm_1_1x3_2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_layer_norm_1_1x3_2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_layer_norm_1_1x3_2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_layer_norm_1_1x3_2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_linear_1x1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_linear_1x1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_linear_1x1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_linear_1x1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_linear_nobias_1x1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_linear_nobias_1x1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_linear_nobias_1x1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_linear_nobias_1x1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_lstm_1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_lstm_1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_lstm_1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_lstm_1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_lstm_8_projected_1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_lstm_8_projected_1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_lstm_8_projected_1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_lstm_8_projected_1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mean_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mean_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mean_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mean_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mlp_3x4x5_1x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mlp_3x4x5_1x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mlp_3x4x5_1x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mlp_3x4x5_1x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mlp_nobias_3x4x5_1x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mlp_nobias_3x4x5_1x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mlp_nobias_3x4x5_1x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_mlp_nobias_3x4x5_1x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_resnet50_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_resnet50_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_resnet50_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_resnet50_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_sum_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_sum_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_sum_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_sum_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_trainable_state_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_trainable_state_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_trainable_state_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_trainable_state_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_unrolled_lstm_1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_unrolled_lstm_1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_unrolled_lstm_1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_unrolled_lstm_1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vanilla_rnn_8_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vanilla_rnn_8_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vanilla_rnn_8_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vanilla_rnn_8_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_ema_eval_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_ema_eval_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_ema_eval_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_ema_eval_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_ema_train_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_ema_train_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_ema_train_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_from_non_distributed_vqvae_ema_train_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_base_batch_norm_1x2x2x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_base_batch_norm_1x2x2x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_base_batch_norm_scale_offset_1x2x2x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_base_batch_norm_scale_offset_1x2x2x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_batch_norm_1x2x2x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_batch_norm_1x2x2x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_batch_norm_scale_offset_1x2x2x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_batch_norm_scale_offset_1x2x2x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_batch_norm_training_1x2x2x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_batch_norm_training_1x2x2x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_bias_3x3x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_bias_3x3x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_cifar10_convnet_2x3_2x2_1x3x3x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv1d_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv1d_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv1d_lstm_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv1d_lstm_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv1d_transpose_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv1d_transpose_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv2d_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv2d_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv2d_lstm_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv2d_lstm_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv2d_transpose_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv2d_transpose_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv3d_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv3d_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv3d_lstm_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv3d_lstm_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv3d_transpose_3x3_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_conv3d_transpose_3x3_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_cross_replica_batch_norm_1x2x2x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_cross_replica_batch_norm_1x2x2x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_dropout_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_dropout_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_ema_2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_ema_2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_embed_100_100_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_embed_100_100_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_group_norm_2_1x3x4_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_group_norm_2_1x3x4_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_gru_1_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_gru_1_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_base_batch_norm_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_base_batch_norm_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_base_batch_norm_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_base_batch_norm_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_base_batch_norm_scale_offset_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_base_batch_norm_scale_offset_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_base_batch_norm_scale_offset_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_base_batch_norm_scale_offset_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_scale_offset_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_scale_offset_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_scale_offset_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_scale_offset_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_training_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_training_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_training_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_batch_norm_training_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_bias_3x3x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_bias_3x3x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_bias_3x3x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_bias_3x3x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_cifar10_convnet_2x3_2x2_1x3x3x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_cifar10_convnet_2x3_2x2_1x3x3x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_lstm_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_lstm_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_lstm_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_lstm_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_transpose_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_transpose_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_transpose_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv1d_transpose_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_lstm_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_lstm_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_lstm_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_lstm_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_transpose_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_transpose_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_transpose_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv2d_transpose_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_lstm_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_lstm_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_lstm_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_lstm_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_transpose_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_transpose_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_transpose_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_conv3d_transpose_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_cross_replica_batch_norm_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_cross_replica_batch_norm_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_cross_replica_batch_norm_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_cross_replica_batch_norm_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_dropout_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_dropout_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_dropout_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_dropout_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_ema_2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_ema_2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_ema_2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_ema_2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_embed_100_100_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_embed_100_100_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_embed_100_100_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_embed_100_100_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_group_norm_2_1x3x4_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_group_norm_2_1x3x4_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_group_norm_2_1x3x4_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_group_norm_2_1x3x4_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_gru_1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_gru_1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_gru_1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_gru_1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_instance_norm_1_1x3_2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_instance_norm_1_1x3_2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_instance_norm_1_1x3_2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_instance_norm_1_1x3_2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_layer_norm_1_1x3_2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_layer_norm_1_1x3_2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_layer_norm_1_1x3_2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_layer_norm_1_1x3_2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_linear_1x1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_linear_1x1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_linear_1x1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_linear_1x1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_linear_nobias_1x1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_linear_nobias_1x1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_linear_nobias_1x1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_linear_nobias_1x1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_lstm_1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_lstm_1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_lstm_1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_lstm_1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_lstm_8_projected_1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_lstm_8_projected_1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_lstm_8_projected_1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_lstm_8_projected_1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mean_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mean_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mean_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mean_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mlp_3x4x5_1x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mlp_3x4x5_1x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mlp_3x4x5_1x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mlp_3x4x5_1x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mlp_nobias_3x4x5_1x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mlp_nobias_3x4x5_1x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mlp_nobias_3x4x5_1x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_mlp_nobias_3x4x5_1x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_resnet50_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_resnet50_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_resnet50_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_resnet50_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_sum_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_sum_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_sum_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_sum_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_trainable_state_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_trainable_state_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_trainable_state_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_trainable_state_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_unrolled_lstm_1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_unrolled_lstm_1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_unrolled_lstm_1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_unrolled_lstm_1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vanilla_rnn_8_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vanilla_rnn_8_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vanilla_rnn_8_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vanilla_rnn_8_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_ema_eval_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_ema_eval_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_ema_eval_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_ema_eval_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_ema_train_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_ema_train_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_ema_train_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_in_replica_context_vqvae_ema_train_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_instance_norm_1_1x3_2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_instance_norm_1_1x3_2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_layer_norm_1_1x3_2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_layer_norm_1_1x3_2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_linear_1x1_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_linear_1x1_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_linear_nobias_1x1_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_linear_nobias_1x1_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_lstm_1_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_lstm_1_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_lstm_8_projected_1_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_lstm_8_projected_1_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_mean_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_mean_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_mlp_3x4x5_1x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_mlp_3x4x5_1x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_mlp_nobias_3x4x5_1x3_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_mlp_nobias_3x4x5_1x3_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_resnet50_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_resnet50_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_sum_2x2_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_sum_2x2_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_trainable_state_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_trainable_state_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_unrolled_lstm_1_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_unrolled_lstm_1_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_vanilla_rnn_8_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_vanilla_rnn_8_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_vqvae_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_vqvae_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_vqvae_ema_eval_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_vqvae_ema_eval_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_vqvae_ema_train_Replicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_restore_on_create_vqvae_ema_train_TpuReplicator
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_base_batch_norm_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_base_batch_norm_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_base_batch_norm_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_base_batch_norm_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_base_batch_norm_scale_offset_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_base_batch_norm_scale_offset_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_base_batch_norm_scale_offset_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_base_batch_norm_scale_offset_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_scale_offset_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_scale_offset_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_scale_offset_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_scale_offset_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_training_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_training_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_training_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_batch_norm_training_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_bias_3x3x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_bias_3x3x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_bias_3x3x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_bias_3x3x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_cifar10_convnet_2x3_2x2_1x3x3x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_cifar10_convnet_2x3_2x2_1x3x3x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_lstm_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_lstm_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_lstm_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_lstm_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_transpose_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_transpose_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_transpose_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv1d_transpose_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_lstm_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_lstm_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_lstm_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_lstm_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_transpose_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_transpose_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_transpose_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv2d_transpose_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_lstm_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_lstm_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_lstm_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_lstm_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_transpose_3x3_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_transpose_3x3_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_transpose_3x3_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_conv3d_transpose_3x3_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_cross_replica_batch_norm_1x2x2x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_cross_replica_batch_norm_1x2x2x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_cross_replica_batch_norm_1x2x2x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_cross_replica_batch_norm_1x2x2x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_dropout_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_dropout_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_dropout_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_dropout_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_ema_2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_ema_2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_ema_2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_ema_2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_embed_100_100_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_embed_100_100_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_embed_100_100_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_embed_100_100_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_group_norm_2_1x3x4_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_group_norm_2_1x3x4_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_group_norm_2_1x3x4_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_group_norm_2_1x3x4_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_gru_1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_gru_1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_gru_1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_gru_1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_instance_norm_1_1x3_2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_instance_norm_1_1x3_2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_instance_norm_1_1x3_2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_instance_norm_1_1x3_2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_layer_norm_1_1x3_2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_layer_norm_1_1x3_2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_layer_norm_1_1x3_2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_layer_norm_1_1x3_2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_linear_1x1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_linear_1x1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_linear_1x1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_linear_1x1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_linear_nobias_1x1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_linear_nobias_1x1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_linear_nobias_1x1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_linear_nobias_1x1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_lstm_1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_lstm_1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_lstm_1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_lstm_1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_lstm_8_projected_1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_lstm_8_projected_1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_lstm_8_projected_1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_lstm_8_projected_1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mean_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mean_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mean_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mean_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mlp_3x4x5_1x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mlp_3x4x5_1x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mlp_3x4x5_1x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mlp_3x4x5_1x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mlp_nobias_3x4x5_1x3_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mlp_nobias_3x4x5_1x3_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mlp_nobias_3x4x5_1x3_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_mlp_nobias_3x4x5_1x3_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_resnet50_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_resnet50_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_resnet50_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_resnet50_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_sum_2x2_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_sum_2x2_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_sum_2x2_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_sum_2x2_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_trainable_state_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_trainable_state_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_trainable_state_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_trainable_state_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_unrolled_lstm_1_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_unrolled_lstm_1_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_unrolled_lstm_1_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_unrolled_lstm_1_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vanilla_rnn_8_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vanilla_rnn_8_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vanilla_rnn_8_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vanilla_rnn_8_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_ema_eval_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_ema_eval_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_ema_eval_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_ema_eval_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_ema_train_Replicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_ema_train_Replicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_ema_train_TpuReplicator_not_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_save_restore_vqvae_ema_train_TpuReplicator_use_function
sonnet/src/conformance/checkpoint_test.py::ReplicatorCheckpointTest::test_session
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_base_batch_norm_1x2x2x3
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_base_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_batch_norm_1x2x2x3
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_batch_norm_training_1x2x2x3
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_bias_3x3x3
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_cifar10_convnet_2x3_2x2_1x3x3x2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv1d_3x3_2x2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv1d_lstm_3x3_2x2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv1d_transpose_3x3_2x2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv2d_3x3_2x2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv2d_lstm_3x3_2x2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv2d_transpose_3x3_2x2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv3d_3x3_2x2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv3d_lstm_3x3_2x2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_conv3d_transpose_3x3_2x2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_cross_replica_batch_norm_1x2x2x3
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_dropout
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_ema_2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_embed_100_100
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_group_norm_2_1x3x4
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_gru_1
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_instance_norm_1_1x3_2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_layer_norm_1_1x3_2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_linear_1x1
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_linear_nobias_1x1
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_lstm_1
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_lstm_8_projected_1
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_mean_2x2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_mlp_3x4x5_1x3
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_mlp_nobias_3x4x5_1x3
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_resnet50
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_sum_2x2
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_trainable_state
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_unrolled_lstm_1
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_vanilla_rnn_8
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_vqvae
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_vqvae_ema_eval
sonnet/src/conformance/copy_test.py::CopyTest::test_copy_vqvae_ema_train
sonnet/src/conformance/copy_test.py::CopyTest::test_session
sonnet/src/conformance/descriptors_test.py::DescriptorsTest::test_coverage
sonnet/src/conformance/descriptors_test.py::DescriptorsTest::test_session
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_session
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv1DLSTM_dynamic_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv1DLSTM_dynamic_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv1DLSTM_not_dynamic_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv1DLSTM_not_dynamic_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv2DLSTM_dynamic_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv2DLSTM_dynamic_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv2DLSTM_not_dynamic_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv2DLSTM_not_dynamic_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv3DLSTM_dynamic_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv3DLSTM_dynamic_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv3DLSTM_not_dynamic_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_Conv3DLSTM_not_dynamic_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_GRU_dynamic_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_GRU_dynamic_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_GRU_not_dynamic_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_GRU_not_dynamic_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_LSTM_dynamic_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_LSTM_dynamic_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_LSTM_not_dynamic_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_LSTM_not_dynamic_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_VanillaRNN_dynamic_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_VanillaRNN_dynamic_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_VanillaRNN_not_dynamic_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_unroll_VanillaRNN_not_dynamic_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_base_batch_norm_1x2x2x3_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_base_batch_norm_1x2x2x3_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_base_batch_norm_scale_offset_1x2x2x3_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_base_batch_norm_scale_offset_1x2x2x3_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_batch_norm_1x2x2x3_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_batch_norm_1x2x2x3_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_batch_norm_scale_offset_1x2x2x3_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_batch_norm_scale_offset_1x2x2x3_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_batch_norm_training_1x2x2x3_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_batch_norm_training_1x2x2x3_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_bias_3x3x3_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_bias_3x3x3_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_cifar10_convnet_2x3_2x2_1x3x3x2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_cifar10_convnet_2x3_2x2_1x3x3x2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv1d_3x3_2x2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv1d_3x3_2x2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv1d_lstm_3x3_2x2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv1d_lstm_3x3_2x2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv1d_transpose_3x3_2x2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv1d_transpose_3x3_2x2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv2d_3x3_2x2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv2d_3x3_2x2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv2d_lstm_3x3_2x2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv2d_lstm_3x3_2x2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv2d_transpose_3x3_2x2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv2d_transpose_3x3_2x2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv3d_3x3_2x2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv3d_3x3_2x2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv3d_lstm_3x3_2x2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv3d_lstm_3x3_2x2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv3d_transpose_3x3_2x2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_conv3d_transpose_3x3_2x2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_cross_replica_batch_norm_1x2x2x3_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_cross_replica_batch_norm_1x2x2x3_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_dropout_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_dropout_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_ema_2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_ema_2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_embed_100_100_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_embed_100_100_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_group_norm_2_1x3x4_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_group_norm_2_1x3x4_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_gru_1_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_gru_1_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_instance_norm_1_1x3_2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_instance_norm_1_1x3_2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_layer_norm_1_1x3_2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_layer_norm_1_1x3_2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_linear_1x1_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_linear_1x1_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_linear_nobias_1x1_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_linear_nobias_1x1_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_lstm_1_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_lstm_1_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_lstm_8_projected_1_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_lstm_8_projected_1_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_mean_2x2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_mean_2x2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_mlp_3x4x5_1x3_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_mlp_3x4x5_1x3_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_mlp_nobias_3x4x5_1x3_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_mlp_nobias_3x4x5_1x3_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_resnet50_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_resnet50_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_sum_2x2_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_sum_2x2_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_trainable_state_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_trainable_state_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_unrolled_lstm_1_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_unrolled_lstm_1_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_vanilla_rnn_8_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_vanilla_rnn_8_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_vqvae_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_vqvae_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_vqvae_ema_eval_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_vqvae_ema_eval_TpuReplicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_vqvae_ema_train_Replicator
sonnet/src/conformance/distribute_test.py::TpuReplicatorTest::test_variable_creation_in_replica_context_vqvae_ema_train_TpuReplicator
sonnet/src/conformance/doctest_test.py::DoctestTest::test_doctest_sonnet
sonnet/src/conformance/doctest_test.py::DoctestTest::test_doctest_sonnet.distribute
sonnet/src/conformance/doctest_test.py::DoctestTest::test_doctest_sonnet.initializers
sonnet/src/conformance/doctest_test.py::DoctestTest::test_doctest_sonnet.mixed_precision
sonnet/src/conformance/doctest_test.py::DoctestTest::test_doctest_sonnet.nets
sonnet/src/conformance/doctest_test.py::DoctestTest::test_doctest_sonnet.nets.resnet
sonnet/src/conformance/doctest_test.py::DoctestTest::test_doctest_sonnet.optimizers
sonnet/src/conformance/doctest_test.py::DoctestTest::test_doctest_sonnet.pad
sonnet/src/conformance/doctest_test.py::DoctestTest::test_doctest_sonnet.regularizers
sonnet/src/conformance/doctest_test.py::DoctestTest::test_session
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.Adam_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.Adam_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.Momentum_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.Momentum_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.RMSProp_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.RMSProp_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.SGD_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_dense_optimizers.SGD_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.Adam_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.Adam_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.Momentum_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.Momentum_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.RMSProp_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.RMSProp_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.SGD_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_optimizer_sparse_optimizers.SGD_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_session
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_BatchNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_BatchNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Bias_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Bias_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv1DTranspose_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv1DTranspose_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv1D_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv1D_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv2DTranspose_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv2DTranspose_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv2D_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv2D_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv3DTranspose_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv3DTranspose_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv3D_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Conv3D_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_CrossReplicaBatchNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_CrossReplicaBatchNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Dropout_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Dropout_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Embed_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Embed_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Flatten_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Flatten_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_GroupNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_GroupNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_InstanceNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_InstanceNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_LayerNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_LayerNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Linear_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Linear_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv1DLSTM, dynamic_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv1DLSTM, dynamic_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv1DLSTM, static_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv1DLSTM, static_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv2DLSTM, dynamic_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv2DLSTM, dynamic_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv2DLSTM, static_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv2DLSTM, static_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv3DLSTM, dynamic_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv3DLSTM, dynamic_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv3DLSTM, static_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(Conv3DLSTM, static_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(GRU, dynamic_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(GRU, dynamic_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(GRU, static_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(GRU, static_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(LSTM, dynamic_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(LSTM, dynamic_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(LSTM, static_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(LSTM, static_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(UnrolledLSTM)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(UnrolledLSTM)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(VanillaRNN, dynamic_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(VanillaRNN, dynamic_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(VanillaRNN, static_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Recurrent(VanillaRNN, static_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Sequential_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_Sequential_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_BatchNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_BatchNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Bias_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Bias_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv1DTranspose_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv1DTranspose_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv1D_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv1D_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv2DTranspose_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv2DTranspose_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv2D_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv2D_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv3DTranspose_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv3DTranspose_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv3D_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Conv3D_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_CrossReplicaBatchNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_CrossReplicaBatchNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Dropout_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Dropout_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Embed_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Embed_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Flatten_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Flatten_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_GroupNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_GroupNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_InstanceNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_InstanceNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_LayerNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_LayerNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Linear_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Linear_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv1DLSTM, dynamic_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv1DLSTM, dynamic_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv1DLSTM, static_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv1DLSTM, static_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv2DLSTM, dynamic_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv2DLSTM, dynamic_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv2DLSTM, static_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv2DLSTM, static_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv3DLSTM, dynamic_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv3DLSTM, dynamic_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv3DLSTM, static_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(Conv3DLSTM, static_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(GRU, dynamic_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(GRU, dynamic_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(GRU, static_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(GRU, static_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(LSTM, dynamic_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(LSTM, dynamic_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(LSTM, static_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(LSTM, static_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(UnrolledLSTM)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(UnrolledLSTM)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(VanillaRNN, dynamic_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(VanillaRNN, dynamic_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(VanillaRNN, static_unroll)_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Recurrent(VanillaRNN, static_unroll)_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Sequential_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_Sequential_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.Cifar10ConvNet_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.Cifar10ConvNet_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.MLP_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.MLP_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.ResNet50_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.ResNet50_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.VectorQuantizerEMA_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.VectorQuantizerEMA_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.VectorQuantizer_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_agnostic_nets.VectorQuantizer_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_BatchNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_BatchNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Bias_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Bias_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv1DTranspose_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv1DTranspose_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv1D_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv1D_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv2DTranspose_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv2DTranspose_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv2D_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv2D_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv3DTranspose_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv3DTranspose_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv3D_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Conv3D_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_CrossReplicaBatchNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_CrossReplicaBatchNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Dropout_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Dropout_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Embed_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Embed_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Flatten_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Flatten_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_GroupNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_GroupNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_InstanceNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_InstanceNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_LayerNorm_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_LayerNorm_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Linear_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Linear_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Sequential_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_Sequential_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.Cifar10ConvNet_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.Cifar10ConvNet_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.MLP_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.MLP_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.ResNet50_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.ResNet50_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.VectorQuantizerEMA_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.VectorQuantizerEMA_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.VectorQuantizer_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_batch_apply_batch_agnostic_nets.VectorQuantizer_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.Cifar10ConvNet_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.Cifar10ConvNet_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.MLP_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.MLP_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.ResNet50_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.ResNet50_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.VectorQuantizerEMA_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.VectorQuantizerEMA_not_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.VectorQuantizer_autograph
sonnet/src/conformance/function_test.py::FunctionTest::test_trace_nets.VectorQuantizer_not_autograph
sonnet/src/conformance/goldens_test.py::CoverageTest::test_all_modules_covered
sonnet/src/conformance/goldens_test.py::CoverageTest::test_session
sonnet/src/conformance/keras_test.py::KerasTest::test_build_with_updating_module
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_BatchNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Bias
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Conv1D
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Conv1DTranspose
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Conv2D
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Conv2DTranspose
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Conv3D
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Conv3DTranspose
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_CrossReplicaBatchNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Dropout
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Embed
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Flatten
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_GroupNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_InstanceNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_LayerNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Linear
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(Conv1DLSTM, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(Conv1DLSTM, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(Conv2DLSTM, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(Conv2DLSTM, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(Conv3DLSTM, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(Conv3DLSTM, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(GRU, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(GRU, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(LSTM, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(LSTM, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(UnrolledLSTM)
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(VanillaRNN, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Recurrent(VanillaRNN, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_Sequential
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_nets.Cifar10ConvNet
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_nets.MLP
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_nets.ResNet50
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_nets.VectorQuantizer
sonnet/src/conformance/keras_test.py::KerasTest::test_build_without_batch_nets.VectorQuantizerEMA
sonnet/src/conformance/keras_test.py::KerasTest::test_from_config
sonnet/src/conformance/keras_test.py::KerasTest::test_keras_layer_inside_sonnet_module
sonnet/src/conformance/keras_test.py::KerasTest::test_layer_adapter_custom_method
sonnet/src/conformance/keras_test.py::KerasTest::test_layer_with_model
sonnet/src/conformance/keras_test.py::KerasTest::test_session
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_BatchNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Bias
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Conv1D
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Conv1DTranspose
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Conv2D
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Conv2DTranspose
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Conv3D
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Conv3DTranspose
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_CrossReplicaBatchNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Dropout
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Embed
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Flatten
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_GroupNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_InstanceNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_LayerNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Linear
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(Conv1DLSTM, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(Conv1DLSTM, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(Conv2DLSTM, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(Conv2DLSTM, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(Conv3DLSTM, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(Conv3DLSTM, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(GRU, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(GRU, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(LSTM, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(LSTM, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(UnrolledLSTM)
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(VanillaRNN, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Recurrent(VanillaRNN, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_Sequential
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_nets.Cifar10ConvNet
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_nets.MLP
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_nets.ResNet50
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_nets.VectorQuantizer
sonnet/src/conformance/keras_test.py::KerasTest::test_sonnet_module_as_layer_nets.VectorQuantizerEMA
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_BatchNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Bias
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Conv1D
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Conv1DTranspose
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Conv2D
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Conv2DTranspose
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Conv3D
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Conv3DTranspose
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_CrossReplicaBatchNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Dropout
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Embed
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Flatten
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_GroupNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_InstanceNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_LayerNorm
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Linear
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(Conv1DLSTM, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(Conv1DLSTM, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(Conv2DLSTM, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(Conv2DLSTM, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(Conv3DLSTM, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(Conv3DLSTM, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(GRU, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(GRU, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(LSTM, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(LSTM, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(UnrolledLSTM)
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(VanillaRNN, dynamic_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Recurrent(VanillaRNN, static_unroll)
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_Sequential
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_nets.Cifar10ConvNet
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_nets.MLP
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_nets.ResNet50
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_nets.VectorQuantizer
sonnet/src/conformance/keras_test.py::KerasTest::test_symbolic_model_nets.VectorQuantizerEMA
sonnet/src/conformance/keras_test.py::KerasTest::test_to_config
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_session
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_BatchNorm_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_BatchNorm_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Bias_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Bias_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv1DTranspose_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv1DTranspose_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv1D_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv1D_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv2DTranspose_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv2DTranspose_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv2D_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv2D_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv3DTranspose_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv3DTranspose_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv3D_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Conv3D_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_CrossReplicaBatchNorm_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_CrossReplicaBatchNorm_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Dropout_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Dropout_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Embed_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Embed_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Flatten_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Flatten_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_GroupNorm_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_GroupNorm_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_InstanceNorm_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_InstanceNorm_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_LayerNorm_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_LayerNorm_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Linear_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Linear_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv1DLSTM, dynamic_unroll)_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv1DLSTM, dynamic_unroll)_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv1DLSTM, static_unroll)_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv1DLSTM, static_unroll)_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv2DLSTM, dynamic_unroll)_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv2DLSTM, dynamic_unroll)_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv2DLSTM, static_unroll)_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv2DLSTM, static_unroll)_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv3DLSTM, dynamic_unroll)_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv3DLSTM, dynamic_unroll)_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv3DLSTM, static_unroll)_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(Conv3DLSTM, static_unroll)_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(GRU, dynamic_unroll)_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(GRU, dynamic_unroll)_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(GRU, static_unroll)_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(GRU, static_unroll)_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(LSTM, dynamic_unroll)_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(LSTM, dynamic_unroll)_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(LSTM, static_unroll)_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(LSTM, static_unroll)_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(UnrolledLSTM)_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(UnrolledLSTM)_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(VanillaRNN, dynamic_unroll)_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(VanillaRNN, dynamic_unroll)_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(VanillaRNN, static_unroll)_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Recurrent(VanillaRNN, static_unroll)_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Sequential_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_Sequential_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.Cifar10ConvNet_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.Cifar10ConvNet_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.MLP_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.MLP_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.ResNet50_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.ResNet50_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.VectorQuantizerEMA_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.VectorQuantizerEMA_not_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.VectorQuantizer_construct_module_in_function
sonnet/src/conformance/optimizer_test.py::OptimizerConformanceTest::test_variable_order_is_constant_nets.VectorQuantizer_not_construct_module_in_function
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_base_batch_norm_1x2x2x3
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_base_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_batch_norm_1x2x2x3
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_batch_norm_training_1x2x2x3
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_bias_3x3x3
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_cifar10_convnet_2x3_2x2_1x3x3x2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv1d_3x3_2x2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv1d_lstm_3x3_2x2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv1d_transpose_3x3_2x2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv2d_3x3_2x2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv2d_lstm_3x3_2x2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv2d_transpose_3x3_2x2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv3d_3x3_2x2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv3d_lstm_3x3_2x2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_conv3d_transpose_3x3_2x2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_cross_replica_batch_norm_1x2x2x3
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_dropout
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_ema_2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_embed_100_100
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_group_norm_2_1x3x4
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_gru_1
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_instance_norm_1_1x3_2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_layer_norm_1_1x3_2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_linear_1x1
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_linear_nobias_1x1
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_lstm_1
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_lstm_8_projected_1
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_mean_2x2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_mlp_3x4x5_1x3
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_mlp_nobias_3x4x5_1x3
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_resnet50
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_sum_2x2
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_trainable_state
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_unrolled_lstm_1
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_vanilla_rnn_8
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_vqvae
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_vqvae_ema_eval
sonnet/src/conformance/pickle_test.py::PickleTest::test_pickle_vqvae_ema_train
sonnet/src/conformance/pickle_test.py::PickleTest::test_session
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_base_batch_norm_1x2x2x3
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_base_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_batch_norm_1x2x2x3
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_batch_norm_training_1x2x2x3
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_bias_3x3x3
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_cifar10_convnet_2x3_2x2_1x3x3x2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv1d_3x3_2x2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv1d_lstm_3x3_2x2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv1d_transpose_3x3_2x2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv2d_3x3_2x2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv2d_lstm_3x3_2x2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv2d_transpose_3x3_2x2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv3d_3x3_2x2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv3d_lstm_3x3_2x2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_conv3d_transpose_3x3_2x2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_cross_replica_batch_norm_1x2x2x3
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_dropout
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_ema_2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_embed_100_100
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_group_norm_2_1x3x4
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_gru_1
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_instance_norm_1_1x3_2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_layer_norm_1_1x3_2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_linear_1x1
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_linear_nobias_1x1
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_lstm_1
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_lstm_8_projected_1
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_mean_2x2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_mlp_3x4x5_1x3
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_mlp_nobias_3x4x5_1x3
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_resnet50
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_sum_2x2
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_trainable_state
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_unrolled_lstm_1
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_vanilla_rnn_8
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_vqvae
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_vqvae_ema_eval
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_save_restore_cycle_vqvae_ema_train
sonnet/src/conformance/saved_model_test.py::SavedModelTest::test_session
sonnet/src/conformance/tensorflow1_test.py::TensorFlow1Test::test_requires_tf2
sonnet/src/conformance/tensorflow1_test.py::TensorFlow1Test::test_session
sonnet/src/conformance/xla_test.py::XLATest::test_compile_base_batch_norm_1x2x2x3
sonnet/src/conformance/xla_test.py::XLATest::test_compile_base_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/xla_test.py::XLATest::test_compile_batch_norm_1x2x2x3
sonnet/src/conformance/xla_test.py::XLATest::test_compile_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/xla_test.py::XLATest::test_compile_batch_norm_training_1x2x2x3
sonnet/src/conformance/xla_test.py::XLATest::test_compile_bias_3x3x3
sonnet/src/conformance/xla_test.py::XLATest::test_compile_cifar10_convnet_2x3_2x2_1x3x3x2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv1d_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv1d_lstm_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv1d_transpose_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv2d_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv2d_lstm_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv2d_transpose_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv3d_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv3d_lstm_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_conv3d_transpose_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_cross_replica_batch_norm_1x2x2x3
sonnet/src/conformance/xla_test.py::XLATest::test_compile_dropout
sonnet/src/conformance/xla_test.py::XLATest::test_compile_ema_2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_embed_100_100
sonnet/src/conformance/xla_test.py::XLATest::test_compile_group_norm_2_1x3x4
sonnet/src/conformance/xla_test.py::XLATest::test_compile_gru_1
sonnet/src/conformance/xla_test.py::XLATest::test_compile_instance_norm_1_1x3_2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_layer_norm_1_1x3_2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_linear_1x1
sonnet/src/conformance/xla_test.py::XLATest::test_compile_linear_nobias_1x1
sonnet/src/conformance/xla_test.py::XLATest::test_compile_lstm_1
sonnet/src/conformance/xla_test.py::XLATest::test_compile_lstm_8_projected_1
sonnet/src/conformance/xla_test.py::XLATest::test_compile_mean_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_mlp_3x4x5_1x3
sonnet/src/conformance/xla_test.py::XLATest::test_compile_mlp_nobias_3x4x5_1x3
sonnet/src/conformance/xla_test.py::XLATest::test_compile_resnet50
sonnet/src/conformance/xla_test.py::XLATest::test_compile_sum_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_compile_trainable_state
sonnet/src/conformance/xla_test.py::XLATest::test_compile_unrolled_lstm_1
sonnet/src/conformance/xla_test.py::XLATest::test_compile_vanilla_rnn_8
sonnet/src/conformance/xla_test.py::XLATest::test_compile_vqvae
sonnet/src/conformance/xla_test.py::XLATest::test_compile_vqvae_ema_eval
sonnet/src/conformance/xla_test.py::XLATest::test_compile_vqvae_ema_train
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_base_batch_norm_1x2x2x3
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_base_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_batch_norm_1x2x2x3
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_batch_norm_scale_offset_1x2x2x3
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_batch_norm_training_1x2x2x3
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_bias_3x3x3
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_cifar10_convnet_2x3_2x2_1x3x3x2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv1d_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv1d_lstm_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv1d_transpose_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv2d_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv2d_lstm_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv2d_transpose_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv3d_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv3d_lstm_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_conv3d_transpose_3x3_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_cross_replica_batch_norm_1x2x2x3
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_dropout
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_ema_2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_embed_100_100
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_group_norm_2_1x3x4
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_gru_1
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_instance_norm_1_1x3_2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_layer_norm_1_1x3_2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_linear_1x1
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_linear_nobias_1x1
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_lstm_1
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_lstm_8_projected_1
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_mean_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_mlp_3x4x5_1x3
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_mlp_nobias_3x4x5_1x3
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_resnet50
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_sum_2x2
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_trainable_state
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_unrolled_lstm_1
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_vanilla_rnn_8
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_vqvae
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_vqvae_ema_eval
sonnet/src/conformance/xla_test.py::XLATest::test_jit_scope_vqvae_ema_train
sonnet/src/conformance/xla_test.py::XLATest::test_session
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_assign_Replicator_assign_add_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_assign_Replicator_assign_add_not_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_assign_Replicator_assign_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_assign_Replicator_assign_not_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_assign_Replicator_assign_sub_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_assign_Replicator_assign_sub_not_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_assign_TpuReplicator_assign_add_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_assign_TpuReplicator_assign_add_not_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_assign_TpuReplicator_assign_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_assign_TpuReplicator_assign_not_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_assign_TpuReplicator_assign_sub_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_assign_TpuReplicator_assign_sub_not_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_read_value_Replicator_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_read_value_Replicator_not_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_read_value_TpuReplicator_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_read_value_TpuReplicator_not_cross_replica
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_session
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_aggregation_default_Replicator_cross_replica_context
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_aggregation_default_Replicator_replica_context
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_aggregation_default_TpuReplicator_cross_replica_context
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_aggregation_default_TpuReplicator_replica_context
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_synchronization_default_Replicator_cross_replica_context
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_synchronization_default_Replicator_replica_context
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_synchronization_default_TpuReplicator_cross_replica_context
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_synchronization_default_TpuReplicator_replica_context
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_trainable_Replicator_not_trainable
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_trainable_Replicator_trainable
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_trainable_TpuReplicator_not_trainable
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_trainable_TpuReplicator_trainable
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_trainable_default_Replicator_cross_replica_context
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_trainable_default_Replicator_replica_context
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_trainable_default_TpuReplicator_cross_replica_context
sonnet/src/distribute/replicator_test.py::ReplicatorTest::test_variable_trainable_default_TpuReplicator_replica_context
sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testDefunBackProp
sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testDifferentSizedImages
sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testFailedModelCreation
sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testModelCreation
sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testModelForwards0
sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testModelForwards1
sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testModelForwards2
sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testModelForwardsFunction0
sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testModelForwardsFunction1
sonnet/src/nets/cifar10_convnet_test.py::ModelTest::testModelForwardsFunction2
sonnet/src/nets/cifar10_convnet_test.py::ModelTest::test_session
sonnet/src/nets/mlp_test.py::MLPTest::test_activate_final0
sonnet/src/nets/mlp_test.py::MLPTest::test_activate_final1
sonnet/src/nets/mlp_test.py::MLPTest::test_activate_final2
sonnet/src/nets/mlp_test.py::MLPTest::test_adds_index_to_layer_names0
sonnet/src/nets/mlp_test.py::MLPTest::test_adds_index_to_layer_names1
sonnet/src/nets/mlp_test.py::MLPTest::test_adds_index_to_layer_names2
sonnet/src/nets/mlp_test.py::MLPTest::test_applies_activation0
sonnet/src/nets/mlp_test.py::MLPTest::test_applies_activation1
sonnet/src/nets/mlp_test.py::MLPTest::test_applies_activation2
sonnet/src/nets/mlp_test.py::MLPTest::test_applies_activation_with_dropout0
sonnet/src/nets/mlp_test.py::MLPTest::test_applies_activation_with_dropout1
sonnet/src/nets/mlp_test.py::MLPTest::test_applies_activation_with_dropout2
sonnet/src/nets/mlp_test.py::MLPTest::test_applies_activation_with_dropout3
sonnet/src/nets/mlp_test.py::MLPTest::test_b_init_when_with_bias_false
sonnet/src/nets/mlp_test.py::MLPTest::test_custom_name
sonnet/src/nets/mlp_test.py::MLPTest::test_default_name
sonnet/src/nets/mlp_test.py::MLPTest::test_dropout_requires_is_training
sonnet/src/nets/mlp_test.py::MLPTest::test_no_dropout_rejects_is_training0
sonnet/src/nets/mlp_test.py::MLPTest::test_no_dropout_rejects_is_training1
sonnet/src/nets/mlp_test.py::MLPTest::test_passes_with_bias_to_layers0
sonnet/src/nets/mlp_test.py::MLPTest::test_passes_with_bias_to_layers1
sonnet/src/nets/mlp_test.py::MLPTest::test_repeat_initializer
sonnet/src/nets/mlp_test.py::MLPTest::test_repr
sonnet/src/nets/mlp_test.py::MLPTest::test_reverse
sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_activate_final0
sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_activate_final1
sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_activation
sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_b_init
sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_custom_name
sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_default_name
sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_override_name
sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_passed_with_bias0
sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_passed_with_bias1
sonnet/src/nets/mlp_test.py::MLPTest::test_reverse_w_init
sonnet/src/nets/mlp_test.py::MLPTest::test_session
sonnet/src/nets/mlp_test.py::MLPTest::test_submodules0
sonnet/src/nets/mlp_test.py::MLPTest::test_submodules1
sonnet/src/nets/mlp_test.py::MLPTest::test_submodules2
sonnet/src/nets/mlp_test.py::MLPTest::test_submodules3
sonnet/src/nets/mlp_test.py::MLPTest::test_submodules4
sonnet/src/nets/mlp_test.py::MLPTest::test_submodules5
sonnet/src/nets/mlp_test.py::MLPTest::test_submodules6
sonnet/src/nets/mlp_test.py::MLPTest::test_submodules7
sonnet/src/nets/mlp_test.py::MLPTest::test_submodules8
sonnet/src/nets/resnet_test.py::ResnetTest::test_error_incorrect_args_block_list0
sonnet/src/nets/resnet_test.py::ResnetTest::test_error_incorrect_args_block_list1
sonnet/src/nets/resnet_test.py::ResnetTest::test_error_incorrect_args_channel_list0
sonnet/src/nets/resnet_test.py::ResnetTest::test_error_incorrect_args_channel_list1
sonnet/src/nets/resnet_test.py::ResnetTest::test_session
sonnet/src/nets/resnet_test.py::ResnetTest::test_simple0
sonnet/src/nets/resnet_test.py::ResnetTest::test_simple1
sonnet/src/nets/resnet_test.py::ResnetTest::test_tf_function0
sonnet/src/nets/resnet_test.py::ResnetTest::test_tf_function1
sonnet/src/nets/vqvae_test.py::VqvaeTest::testConstruct0
sonnet/src/nets/vqvae_test.py::VqvaeTest::testConstruct1
sonnet/src/nets/vqvae_test.py::VqvaeTest::testEmaUpdating0
sonnet/src/nets/vqvae_test.py::VqvaeTest::testEmaUpdating1
sonnet/src/nets/vqvae_test.py::VqvaeTest::testNoneBatch0
sonnet/src/nets/vqvae_test.py::VqvaeTest::testNoneBatch1
sonnet/src/nets/vqvae_test.py::VqvaeTest::testShapeChecking0
sonnet/src/nets/vqvae_test.py::VqvaeTest::testShapeChecking1
sonnet/src/nets/vqvae_test.py::VqvaeTest::test_session
sonnet/src/nets/dnc/control_test.py::CoreTest::testShape0
sonnet/src/nets/dnc/control_test.py::CoreTest::testShape1
sonnet/src/nets/dnc/control_test.py::CoreTest::test_session
sonnet/src/nets/dnc/control_test.py::FeedForwardTest::testShape
sonnet/src/nets/dnc/control_test.py::FeedForwardTest::testValues
sonnet/src/nets/dnc/control_test.py::FeedForwardTest::test_session
sonnet/src/nets/dnc/control_test.py::DeepCore::testShape0
sonnet/src/nets/dnc/control_test.py::DeepCore::testShape1
sonnet/src/nets/dnc/control_test.py::DeepCore::testShape2
sonnet/src/nets/dnc/control_test.py::DeepCore::testShape3
sonnet/src/nets/dnc/control_test.py::DeepCore::test_session
sonnet/src/nets/dnc/read_test.py::ReadTest::testShape
sonnet/src/nets/dnc/read_test.py::ReadTest::testValues
sonnet/src/nets/dnc/read_test.py::ReadTest::test_session
sonnet/src/nets/dnc/util_test.py::SegmentDimTest::testInvalidDims
sonnet/src/nets/dnc/util_test.py::SegmentDimTest::testShape0
sonnet/src/nets/dnc/util_test.py::SegmentDimTest::testShape1
sonnet/src/nets/dnc/util_test.py::SegmentDimTest::testShape2
sonnet/src/nets/dnc/util_test.py::SegmentDimTest::testShape3
sonnet/src/nets/dnc/util_test.py::SegmentDimTest::testShape4
sonnet/src/nets/dnc/util_test.py::SegmentDimTest::testShapeNegative0
sonnet/src/nets/dnc/util_test.py::SegmentDimTest::testShapeNegative1
sonnet/src/nets/dnc/util_test.py::SegmentDimTest::testShapeNegative2
sonnet/src/nets/dnc/util_test.py::SegmentDimTest::testShapeNegative3
sonnet/src/nets/dnc/util_test.py::SegmentDimTest::testShapeNegative4
sonnet/src/nets/dnc/util_test.py::SegmentDimTest::testValues
sonnet/src/nets/dnc/util_test.py::SegmentDimTest::test_session
sonnet/src/nets/dnc/util_test.py::BatchInvertPermutationTest::testCorrectOutput
sonnet/src/nets/dnc/util_test.py::BatchInvertPermutationTest::test_session
sonnet/src/nets/dnc/util_test.py::BatchGatherTest::testCorrectOutput
sonnet/src/nets/dnc/util_test.py::BatchGatherTest::test_session
sonnet/src/nets/dnc/util_test.py::LinearTest::testDifferentOutputSizeBreaks
sonnet/src/nets/dnc/util_test.py::LinearTest::testLinearOutputOneModule
sonnet/src/nets/dnc/util_test.py::LinearTest::testLinearOutputTwoModules
sonnet/src/nets/dnc/util_test.py::LinearTest::testListMustBeLengthTwo0
sonnet/src/nets/dnc/util_test.py::LinearTest::testListMustBeLengthTwo1
sonnet/src/nets/dnc/util_test.py::LinearTest::testNonMatchingStructureBreaks0
sonnet/src/nets/dnc/util_test.py::LinearTest::testNonMatchingStructureBreaks1
sonnet/src/nets/dnc/util_test.py::LinearTest::test_session
sonnet/src/nets/dnc/write_test.py::EraseRowsTest::testShape
sonnet/src/nets/dnc/write_test.py::EraseRowsTest::testValues
sonnet/src/nets/dnc/write_test.py::EraseRowsTest::test_session
sonnet/src/nets/dnc/write_test.py::EraseTest::testShape
sonnet/src/nets/dnc/write_test.py::EraseTest::testValues
sonnet/src/nets/dnc/write_test.py::EraseTest::test_session
sonnet/src/nets/dnc/write_test.py::EraseAndWriteTest::testShape
sonnet/src/nets/dnc/write_test.py::EraseAndWriteTest::testValues
sonnet/src/nets/dnc/write_test.py::EraseAndWriteTest::test_session
sonnet/src/nets/dnc/write_test.py::AdditiveWriteTest::testShape
sonnet/src/nets/dnc/write_test.py::AdditiveWriteTest::testValues
sonnet/src/nets/dnc/write_test.py::AdditiveWriteTest::test_session
sonnet/src/optimizers/adam_test.py::AdamTest::testAllUpdatesNone
sonnet/src/optimizers/adam_test.py::AdamTest::testAuxVariablesColocatedWithOriginal
sonnet/src/optimizers/adam_test.py::AdamTest::testDense
sonnet/src/optimizers/adam_test.py::AdamTest::testDifferentLengthUpdatesParams
sonnet/src/optimizers/adam_test.py::AdamTest::testEmptyParams
sonnet/src/optimizers/adam_test.py::AdamTest::testHyperParamDTypeConversion
sonnet/src/optimizers/adam_test.py::AdamTest::testInconsistentDTypes
sonnet/src/optimizers/adam_test.py::AdamTest::testNoneUpdate
sonnet/src/optimizers/adam_test.py::AdamTest::testSparse
sonnet/src/optimizers/adam_test.py::AdamTest::testUnsuppportedStrategyError
sonnet/src/optimizers/adam_test.py::AdamTest::testVariableHyperParams
sonnet/src/optimizers/adam_test.py::AdamTest::test_session
sonnet/src/optimizers/adam_test.py::ReferenceAdamTest::testAllUpdatesNone
sonnet/src/optimizers/adam_test.py::ReferenceAdamTest::testDifferentLengthUpdatesParams
sonnet/src/optimizers/adam_test.py::ReferenceAdamTest::testEmptyParams
sonnet/src/optimizers/adam_test.py::ReferenceAdamTest::testInconsistentDTypes
sonnet/src/optimizers/adam_test.py::ReferenceAdamTest::testNoneUpdate
sonnet/src/optimizers/adam_test.py::ReferenceAdamTest::testUnsuppportedStrategyError
sonnet/src/optimizers/adam_test.py::ReferenceAdamTest::test_session
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.001,momentum=0.2,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.001,momentum=0.2,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.001,momentum=0.5,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.001,momentum=0.5,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.001,momentum=0.9,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.001,momentum=0.9,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.01,momentum=0.2,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.01,momentum=0.2,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.01,momentum=0.5,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.01,momentum=0.5,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.01,momentum=0.9,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.01,momentum=0.9,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.1,momentum=0.2,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.1,momentum=0.2,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.1,momentum=0.5,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.1,momentum=0.5,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.1,momentum=0.9,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=28_learning_rate=0.1,momentum=0.9,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.001,momentum=0.2,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.001,momentum=0.2,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.001,momentum=0.5,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.001,momentum=0.5,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.001,momentum=0.9,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.001,momentum=0.9,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.01,momentum=0.2,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.01,momentum=0.2,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.01,momentum=0.5,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.01,momentum=0.5,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.01,momentum=0.9,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.01,momentum=0.9,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.1,momentum=0.2,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.1,momentum=0.2,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.1,momentum=0.5,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.1,momentum=0.5,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.1,momentum=0.9,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=2_learning_rate=0.1,momentum=0.9,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.001,momentum=0.2,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.001,momentum=0.2,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.001,momentum=0.5,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.001,momentum=0.5,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.001,momentum=0.9,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.001,momentum=0.9,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.01,momentum=0.2,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.01,momentum=0.2,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.01,momentum=0.5,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.01,momentum=0.5,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.01,momentum=0.9,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.01,momentum=0.9,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.1,momentum=0.2,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.1,momentum=0.2,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.1,momentum=0.5,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.1,momentum=0.5,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.1,momentum=0.9,use_nesterov=False
sonnet/src/optimizers/momentum_test.py::ComparisonTest::testComparingSonnetAndTensorFlow_seed=90_learning_rate=0.1,momentum=0.9,use_nesterov=True
sonnet/src/optimizers/momentum_test.py::ComparisonTest::test_session
sonnet/src/optimizers/momentum_test.py::MomentumTest::testAllUpdatesNone
sonnet/src/optimizers/momentum_test.py::MomentumTest::testAuxVariablesColocatedWithOriginal
sonnet/src/optimizers/momentum_test.py::MomentumTest::testDense
sonnet/src/optimizers/momentum_test.py::MomentumTest::testDenseNesterov
sonnet/src/optimizers/momentum_test.py::MomentumTest::testDifferentLengthUpdatesParams
sonnet/src/optimizers/momentum_test.py::MomentumTest::testEmptyParams
sonnet/src/optimizers/momentum_test.py::MomentumTest::testHyperParamDTypeConversion
sonnet/src/optimizers/momentum_test.py::MomentumTest::testInconsistentDTypes
sonnet/src/optimizers/momentum_test.py::MomentumTest::testNoneUpdate
sonnet/src/optimizers/momentum_test.py::MomentumTest::testSparse
sonnet/src/optimizers/momentum_test.py::MomentumTest::testSparseNesterov
sonnet/src/optimizers/momentum_test.py::MomentumTest::testUnsuppportedStrategyError
sonnet/src/optimizers/momentum_test.py::MomentumTest::testVariableHyperParams
sonnet/src/optimizers/momentum_test.py::MomentumTest::test_session
sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testAllUpdatesNone
sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testAuxVariablesColocatedWithOriginal
sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testDense
sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testDenseNesterov
sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testDifferentLengthUpdatesParams
sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testEmptyParams
sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testHyperParamDTypeConversion
sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testInconsistentDTypes
sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testNoneUpdate
sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testSparse
sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testSparseNesterov
sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testUnsuppportedStrategyError
sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::testVariableHyperParams
sonnet/src/optimizers/momentum_test.py::ReferenceMomentumTest::test_session
sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testAllUpdatesNone
sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testAuxVariablesColocatedWithOriginal
sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testDense
sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testDenseCentered
sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testDifferentLengthUpdatesParams
sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testEmptyParams
sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testHyperParamDTypeConversion
sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testInconsistentDTypes
sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testNoneUpdate
sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testSparse
sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testSparseCentered
sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testUnsuppportedStrategyError
sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::testVariableHyperParams
sonnet/src/optimizers/rmsprop_test.py::RMSPropTest::test_session
sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testAllUpdatesNone
sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testAuxVariablesColocatedWithOriginal
sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testDense
sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testDenseCentered
sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testDifferentLengthUpdatesParams
sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testEmptyParams
sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testHyperParamDTypeConversion
sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testInconsistentDTypes
sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testNoneUpdate
sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testSparse
sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testSparseCentered
sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testUnsuppportedStrategyError
sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::testVariableHyperParams
sonnet/src/optimizers/rmsprop_test.py::ReferenceRMSPropTest::test_session
sonnet/src/optimizers/sgd_test.py::SGDTest::testAllUpdatesNone
sonnet/src/optimizers/sgd_test.py::SGDTest::testDense
sonnet/src/optimizers/sgd_test.py::SGDTest::testDifferentLengthUpdatesParams
sonnet/src/optimizers/sgd_test.py::SGDTest::testEmptyParams
sonnet/src/optimizers/sgd_test.py::SGDTest::testInconsistentDTypes
sonnet/src/optimizers/sgd_test.py::SGDTest::testLearningRateDTypeConversion
sonnet/src/optimizers/sgd_test.py::SGDTest::testNoneUpdate
sonnet/src/optimizers/sgd_test.py::SGDTest::testSparse
sonnet/src/optimizers/sgd_test.py::SGDTest::testUnsuppportedStrategyError
sonnet/src/optimizers/sgd_test.py::SGDTest::testVariableLearningRate
sonnet/src/optimizers/sgd_test.py::SGDTest::test_session
sonnet/src/optimizers/sgd_test.py::ReferenceSGDTest::testAllUpdatesNone
sonnet/src/optimizers/sgd_test.py::ReferenceSGDTest::testDense
sonnet/src/optimizers/sgd_test.py::ReferenceSGDTest::testDifferentLengthUpdatesParams
sonnet/src/optimizers/sgd_test.py::ReferenceSGDTest::testEmptyParams
sonnet/src/optimizers/sgd_test.py::ReferenceSGDTest::testInconsistentDTypes
sonnet/src/optimizers/sgd_test.py::ReferenceSGDTest::testLearningRateDTypeConversion
sonnet/src/optimizers/sgd_test.py::ReferenceSGDTest::testNoneUpdate
sonnet/src/optimizers/sgd_test.py::ReferenceSGDTest::testSparse
sonnet/src/optimizers/sgd_test.py::ReferenceSGDTest::testUnsuppportedStrategyError
sonnet/src/optimizers/sgd_test.py::ReferenceSGDTest::testVariableLearningRate
sonnet/src/optimizers/sgd_test.py::ReferenceSGDTest::test_session

