MCPcopy Create free account

hub / github.com/google-deepmind/chex / functions

Functions578 in github.com/google-deepmind/chex

⨍ Functions 578◇ Types & classes 103↳ Endpoints 2
Methodtest_docstring_example
(self)
chex/_src/asserts_chexify_test.py:220
Methodtest_docstring_examples
(self)
chex/_src/dimensions_test.py:36
Methodtest_easy
(self, x_shape, seed)
chex/_src/asserts_test.py:1807
Methodtest_easy_with_stop_gradient
(self, x_shape, seed)
chex/_src/asserts_test.py:1813
Methodtest_ellipsis_should_fail
(self, array, expected_shape)
chex/_src/asserts_test.py:530
Methodtest_ellipsis_should_pass
(self, arrays, expected_size)
chex/_src/asserts_test.py:368
Methodtest_ellipsis_should_pass
(self, array, expected_shape)
chex/_src/asserts_test.py:514
Methodtest_emplace
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:674
Methodtest_emplace
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:696
Methodtest_enable_disable_asserts
(self)
chex/_src/asserts_test.py:62
Methodtest_equal_shape_at_dims_should_fail
(self, shapes, dims)
chex/_src/asserts_test.py:430
Methodtest_equal_shape_at_dims_should_pass
(self, shapes, dims)
chex/_src/asserts_test.py:420
Methodtest_equal_shape_prefix_should_fail
(self, arrays)
chex/_src/asserts_test.py:407
Methodtest_equal_shape_prefix_should_pass
(self, arrays)
chex/_src/asserts_test.py:399
Methodtest_equal_shape_should_fail
(self, arrays)
chex/_src/asserts_test.py:380
Methodtest_equal_shape_should_pass
(self, arrays)
chex/_src/asserts_test.py:391
Methodtest_equal_size_should_fail
(self, arrays)
chex/_src/asserts_test.py:304
Methodtest_equal_size_should_pass
(self, arrays)
chex/_src/asserts_test.py:316
Methodtest_equivalence
(fn)
chex/_src/fake_test.py:377
Methodtest_error
(self)
chex/_src/variants_test.py:256
Methodtest_failure
(self)
chex/_src/variants_test.py:252
Methodtest_failure
(self)
chex/_src/variants_test.py:285
Methodtest_failure
(self)
chex/_src/variants_test.py:313
Methodtest_fake_jit
(self, fake_kwargs, is_jitted)
chex/_src/fake_test.py:114
Methodtest_fake_parallel_axis
(self)
chex/_src/fake_test.py:412
Methodtest_fake_pmap_
(self, is_pmapped, jit_result)
chex/_src/fake_test.py:135
Methodtest_fake_pmap_axis_name
(self)
chex/_src/fake_test.py:152
Methodtest_flatten_fail
(self, named_shape, error_message)
chex/_src/dimensions_test.py:140
Methodtest_flatten_is_leaf
(self, is_mappable)
chex/_src/dataclass_test.py:610
Methodtest_flatten_ok
(self, named_shape, expected_shape)
chex/_src/dimensions_test.py:123
Methodtest_flatten_respects_post_init
(self)
chex/_src/dataclass_test.py:729
Methodtest_flatten_roundtrip_ordering
(self, dcls)
chex/_src/dataclass_test.py:722
Methodtest_format
(self, include_default_msg, include_custom_msg, exc_type)
chex/_src/asserts_internal_test.py:59
Methodtest_frozen_dataclass_raise_error
(self)
chex/_src/dataclass_test.py:483
Methodtest_generic_dataclass
(self, mappable)
chex/_src/dataclass_test.py:668
Methodtest_get_and_set_state
(self, frozen)
chex/_src/dataclass_test.py:495
Methodtest_get_exception
(self, k, e, m)
chex/_src/dimensions_test.py:89
Methodtest_get_literals
(self)
chex/_src/dimensions_test.py:67
Methodtest_get_wildcard
(self)
chex/_src/dimensions_test.py:60
Methodtest_gpu_assert
(self)
chex/_src/asserts_test.py:1757
Methodtest_hard
(self, x_shape, seed)
chex/_src/asserts_test.py:1819
Methodtest_hard_with_stop_gradient
(self, x_shape, seed)
chex/_src/asserts_test.py:1841
Methodtest_ignore_argnums
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:703
Methodtest_ignores_explicitly_allowed_cpu_computation
(self)
chex/_src/restrict_backends_test.py:81
Methodtest_ignores_explicitly_allowed_tpu_computation
(self)
chex/_src/restrict_backends_test.py:91
Methodtest_ignores_implicitly_allowed_cpu_computation
(self)
chex/_src/restrict_backends_test.py:86
Methodtest_ignores_implicitly_allowed_tpu_computation
(self)
chex/_src/restrict_backends_test.py:96
Methodtest_import
(self)
chex/chex_test.py:24
Methodtest_incorrect_ordering
(self)
chex/_src/asserts_test.py:177
Methodtest_incorrect_wrapping_order_all_variants
(self)
chex/_src/variants_test.py:402
Methodtest_incorrect_wrapping_order_named_all_variants
(self)
chex/_src/variants_test.py:386
Methodtest_incorrect_wrapping_order_named_some_variants
(self)
chex/_src/variants_test.py:394
Methodtest_incorrect_wrapping_order_some_variants
(self)
chex/_src/variants_test.py:410
Methodtest_inheritance
(self, frozen)
chex/_src/dataclass_test.py:547
Methodtest_inheritance
(self, base_classes)
chex/_src/variants_test.py:349
Methodtest_inheritance_from_empty_frozen_base
(self)
chex/_src/dataclass_test.py:565
Methodtest_inheritance_is_possible_thanks_to_kw_only
(self)
chex/_src/dataclass_test.py:460
Methodtest_is_traceable
(self)
chex/_src/asserts_internal_test.py:30
Methodtest_keys_and_values_override
(self, frozen)
chex/_src/dataclass_test.py:747
Methodtest_keys_and_values_type
(self, frozen)
chex/_src/dataclass_test.py:735
Methodtest_leaf_shape_should_fail_wrong_length
(self)
chex/_src/asserts_test.py:1586
Methodtest_log_abs_fn_jitted
Tests simple jit transformation.
chex/_src/asserts_chexify_test.py:456
Methodtest_log_abs_fn_jitted_nested_call
Tests nested jit transforms (calling).
chex/_src/asserts_chexify_test.py:509
Methodtest_log_abs_fn_jitted_nested_wrap
Tests nested jit transforms (wrapping).
chex/_src/asserts_chexify_test.py:480
Methodtest_log_abs_fn_jitted_vmapped
Tests vmap transform.
chex/_src/asserts_chexify_test.py:562
Methodtest_log_abs_fn_pmapped
Tests pmap transform.
chex/_src/asserts_chexify_test.py:537
Methodtest_many_variants
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:561
Methodtest_mappable_eq_override
(self)
chex/_src/dataclass_test.py:678
Methodtest_multiple_ellipses
(self, array, expected_shape)
chex/_src/asserts_test.py:544
Functiontest_named_scalar_parameters
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:147
Functiontest_ndarray_named_parameters
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:182
Functiontest_ndarray_parameters
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:168
Methodtest_nested_chexification
Tests nested wrapping.
chex/_src/asserts_chexify_test.py:132
Methodtest_nested_functions
(self)
chex/_src/asserts_test.py:233
Methodtest_no_variants
(self)
chex/_src/variants_test.py:441
Methodtest_noop
(self)
chex/_src/variants_test.py:425
Methodtest_noop
(self)
chex/_src/variants_test.py:448
Methodtest_not_less_than
(self, devtype)
chex/_src/asserts_test.py:1738
Methodtest_on_call_of_transformed_function
(self)
chex/_src/fake_test.py:445
Methodtest_parallel_ops_equivalence
Test equivalence between parallel operations using pmap and vmap.
chex/_src/fake_test.py:371
Methodtest_partial_python_fn
(self)
chex/_src/asserts_chexify_test.py:281
Methodtest_pmap
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:752
Methodtest_pmap_and_jit
(self, fake_kwargs, is_pmapped, is_jitted)
chex/_src/fake_test.py:184
Methodtest_pmap_in_axes
(self)
chex/_src/variants_test.py:893
Methodtest_pmap_static_argnums
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:822
Methodtest_pmap_static_argnums_zero
(self)
chex/_src/variants_test.py:860
Methodtest_pmap_vmapped_fn
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:781
Methodtest_pmap_with_complex_static_broadcasted_object
(self, static_argnums)
chex/_src/fake_test.py:268
Methodtest_pmap_wrong_axis_size
(self)
chex/_src/variants_test.py:925
Methodtest_product
(self)
chex/_src/variants_test.py:205
Methodtest_pytypes_pass
(self)
chex/_src/asserts_test.py:358
Methodtest_pytypes_pass
(self, shape)
chex/_src/asserts_test.py:496
Methodtest_raises_if_contradictory_restrictions_specified
(self)
chex/_src/restrict_backends_test.py:106
Methodtest_raises_if_no_restrictions_specified
(self)
chex/_src/restrict_backends_test.py:101
Methodtest_rank_should_fail_array_expectations
(self)
chex/_src/asserts_test.py:583
Methodtest_rank_should_fail_single
(self, array, rank)
chex/_src/asserts_test.py:605
Methodtest_rank_should_fail_wrong_expectation_structure
(self)
chex/_src/asserts_test.py:589
Methodtest_rank_should_fail_wrong_length
(self, array, rank)
chex/_src/asserts_test.py:627
Methodtest_rank_should_pass_multiple_options
(self, arrays, ranks)
chex/_src/asserts_test.py:657
Methodtest_rank_should_pass_repeated_input
(self, array, rank)
chex/_src/asserts_test.py:648
← previousnext →401–500 of 578, ranked by callers