MCPcopy Create free account

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

Functions578 in github.com/google-deepmind/chex

Methodtest_rank_should_pass_single_input
(self, array, rank)
chex/_src/asserts_test.py:639
Methodtest_redefined_traced_function
(self)
chex/_src/asserts_test.py:204
Methodtest_roundtrip_serialization
(self, serialization_lib, dcls)
chex/_src/dataclass_test.py:703
Methodtest_scalar
(self)
chex/_src/asserts_test.py:252
Methodtest_scalar_in
(self)
chex/_src/asserts_test.py:278
Methodtest_scalar_in_excluded
(self)
chex/_src/asserts_test.py:287
Methodtest_scalar_negative
(self)
chex/_src/asserts_test.py:272
Methodtest_scalar_non_negative
(self)
chex/_src/asserts_test.py:265
Functiontest_scalar_parameters
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:138
Methodtest_scalar_positive
(self)
chex/_src/asserts_test.py:259
Methodtest_set_exception
(self, k, v, e, m)
chex/_src/dimensions_test.py:80
Methodtest_set_n_cpu_devices
(self)
chex/_src/fake_set_n_cpu_devices_test.py:29
Methodtest_set_ok
(self, k, v, shape)
chex/_src/dimensions_test.py:48
Methodtest_set_wildcard
(self)
chex/_src/dimensions_test.py:53
Methodtest_shape_should_fail
(self, arrays, shapes)
chex/_src/asserts_test.py:447
Methodtest_shape_should_fail_wrong_length
(self, arrays, shapes)
chex/_src/asserts_test.py:464
Methodtest_shape_should_pass
(self, arrays, shapes)
chex/_src/asserts_test.py:484
Methodtest_shapes_are_broadcastable
(self, shape_a, shape_b)
chex/_src/asserts_test.py:566
Methodtest_shapes_are_not_broadcastable
(self, shape_a, shape_b)
chex/_src/asserts_test.py:576
Methodtest_should_pass
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:337
Methodtest_should_pass
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:367
Methodtest_should_pass
(self)
chex/_src/variants_test.py:371
Methodtest_size_fail_wildcard
(self, names)
chex/_src/dimensions_test.py:108
Methodtest_size_ok
(self, names, shape, expected_size)
chex/_src/dimensions_test.py:98
Methodtest_size_should_fail
(self, arrays, sizes)
chex/_src/asserts_test.py:330
Methodtest_size_should_fail_wrong_length
(self, arrays, sizes)
chex/_src/asserts_test.py:341
Methodtest_size_should_pass
(self, arrays, sizes)
chex/_src/asserts_test.py:354
Methodtest_static_assertion
(self)
chex/_src/asserts_chexify_test.py:100
Methodtest_tpu_assert
(self)
chex/_src/asserts_test.py:1778
Methodtest_tree_all_finite_passes_finite
(self)
chex/_src/asserts_test.py:923
Methodtest_tree_all_finite_should_fail_inf
(self)
chex/_src/asserts_test.py:928
Methodtest_tree_flatten_with_keys
(self)
chex/_src/dataclass_test.py:372
Methodtest_tree_map_with_keys
(self)
chex/_src/dataclass_test.py:387
Methodtest_tree_map_with_keys_traversal_order
(self)
chex/_src/dataclass_test.py:407
Methodtest_tuple_conversion
(self)
chex/_src/dataclass_test.py:518
Methodtest_tuple_rev_conversion
(self, frozen)
chex/_src/dataclass_test.py:536
Methodtest_type_should_fail_array
(self, array, dtype, wrong_type)
chex/_src/asserts_test.py:687
Methodtest_type_should_fail_mixed
(self)
chex/_src/asserts_test.py:714
Methodtest_type_should_fail_scalar
(self, scalars, wrong_type)
chex/_src/asserts_test.py:674
Methodtest_type_should_fail_wrong_length
(self, array, wrong_type)
chex/_src/asserts_test.py:736
Methodtest_type_should_pass_array
(self, array, dtype, expected_type)
chex/_src/asserts_test.py:710
Methodtest_type_should_pass_mixed
(self)
chex/_src/asserts_test.py:724
Methodtest_type_should_pass_scalar
(self, array, expected_type)
chex/_src/asserts_test.py:700
Methodtest_unexpected_kwargs
(self)
chex/_src/dataclass_test.py:507
Methodtest_unfrozen_dataclass_is_mutable
(self)
chex/_src/dataclass_test.py:474
Methodtest_uninspected_checks
(self)
chex/_src/asserts_chexify_test.py:189
Methodtest_unknown_argument
(self)
chex/_src/variants_test.py:461
Methodtest_unsupported_device
(self)
chex/_src/asserts_test.py:1753
Methodtest_unused_variant
(self)
chex/_src/variants_test.py:428
Methodtest_useful_errors
(self)
chex/_src/variants_test.py:271
Methodtest_useful_failure
(self)
chex/_src/variants_test.py:293
Methodtest_useful_failures
(self)
chex/_src/variants_test.py:266
Methodtest_var_type
(self)
chex/_src/variants_test.py:477
Methodtest_var_type_fetch
(self)
chex/_src/variants_test.py:481
Methodtest_vmapped_fn_named_params
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:575
Methodtest_warn_deprecated_function
(self)
chex/_src/warnings_test.py:63
Methodtest_warn_deprecated_function_no_replacement
(self)
chex/_src/warnings_test.py:53
Methodtest_warn_only_n_pos_args_in_future
(self)
chex/_src/warnings_test.py:47
Methodtest_with_default_params
(self, fake_pmap, fake_jit)
chex/_src/fake_test.py:344
Functiontest_with_kwargs
(self)
chex/_src/variants_test.py:128
Methodtest_with_kwargs
(self, fake_pmap, fake_jit)
chex/_src/fake_test.py:207
Functiontest_with_ndarray_args
(self)
chex/_src/variants_test.py:155
Methodtest_with_partial
(self, fake_pmap, fake_jit)
chex/_src/fake_test.py:317
Functiontest_with_scalar_args
(self)
chex/_src/variants_test.py:112
Methodtest_with_static_broadcasted_argnums
(self, fake_pmap, static_argnums)
chex/_src/fake_test.py:229
Methodtest_wrong_base_class
(self)
chex/_src/variants_test.py:316
Methodtest_wrong_order_of_wrapping
(self)
chex/_src/asserts_chexify_test.py:297
Functiontree_error_msg_fn
(l_1: _ai.TLeaf, l_2: _ai.TLeaf, path: str, i_1: int, i_2: int)
chex/_src/asserts.py:1426
Methodunraisablehook
(unraisable)
chex/_src/asserts_chexify_test.py:201
Functionvariants
Decorates a test to expose Chex variants. The decorated test has access to a decorator called ``self.variant``, which may be applied to functions
chex/_src/variants.py:257
Methodvisit_without_dicts
(x)
chex/_src/dataclass_test.py:339
Methodvmapped_fn
(arg_0, arg_1)
chex/_src/variants_test.py:798
Functionwarn_deprecated_function
A decorator to mark a function definition as deprecated. Example usage: >>> @functools.partial(chex.warn_deprecated_function, replacement='g')
chex/_src/warnings.py:61
Functionwarn_only_n_pos_args_in_future
Warns if more than ``n`` positional arguments are passed to ``fun``. For instance: >>> @functools.partial(chex.warn_only_n_pos_args_in_future, n=
chex/_src/warnings.py:22
Functionwith_jittable_assertions
An alias for `chexify` (see the docs).
chex/_src/asserts_chexify.py:244
Functionwrapper
(*args, **kwargs)
chex/_src/warnings.py:43
Functionwrapper
(backend, *args, **kwargs)
chex/_src/restrict_backends.py:47
Functionwrapper
(*args, **kwargs)
chex/_src/variants.py:361
← previous501–578 of 578, ranked by callers