Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/google-deepmind/chex
/ functions
Functions
578 in github.com/google-deepmind/chex
⨍
Functions
578
◇
Types & classes
103
↳
Endpoints
2
Method
test_rank_should_pass_single_input
(self, array, rank)
chex/_src/asserts_test.py:639
Method
test_redefined_traced_function
(self)
chex/_src/asserts_test.py:204
Method
test_roundtrip_serialization
(self, serialization_lib, dcls)
chex/_src/dataclass_test.py:703
Method
test_scalar
(self)
chex/_src/asserts_test.py:252
Method
test_scalar_in
(self)
chex/_src/asserts_test.py:278
Method
test_scalar_in_excluded
(self)
chex/_src/asserts_test.py:287
Method
test_scalar_negative
(self)
chex/_src/asserts_test.py:272
Method
test_scalar_non_negative
(self)
chex/_src/asserts_test.py:265
Function
test_scalar_parameters
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:138
Method
test_scalar_positive
(self)
chex/_src/asserts_test.py:259
Method
test_set_exception
(self, k, v, e, m)
chex/_src/dimensions_test.py:80
Method
test_set_n_cpu_devices
(self)
chex/_src/fake_set_n_cpu_devices_test.py:29
Method
test_set_ok
(self, k, v, shape)
chex/_src/dimensions_test.py:48
Method
test_set_wildcard
(self)
chex/_src/dimensions_test.py:53
Method
test_shape_should_fail
(self, arrays, shapes)
chex/_src/asserts_test.py:447
Method
test_shape_should_fail_wrong_length
(self, arrays, shapes)
chex/_src/asserts_test.py:464
Method
test_shape_should_pass
(self, arrays, shapes)
chex/_src/asserts_test.py:484
Method
test_shapes_are_broadcastable
(self, shape_a, shape_b)
chex/_src/asserts_test.py:566
Method
test_shapes_are_not_broadcastable
(self, shape_a, shape_b)
chex/_src/asserts_test.py:576
Method
test_should_pass
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:337
Method
test_should_pass
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:367
Method
test_should_pass
(self)
chex/_src/variants_test.py:371
Method
test_size_fail_wildcard
(self, names)
chex/_src/dimensions_test.py:108
Method
test_size_ok
(self, names, shape, expected_size)
chex/_src/dimensions_test.py:98
Method
test_size_should_fail
(self, arrays, sizes)
chex/_src/asserts_test.py:330
Method
test_size_should_fail_wrong_length
(self, arrays, sizes)
chex/_src/asserts_test.py:341
Method
test_size_should_pass
(self, arrays, sizes)
chex/_src/asserts_test.py:354
Method
test_static_assertion
(self)
chex/_src/asserts_chexify_test.py:100
Method
test_tpu_assert
(self)
chex/_src/asserts_test.py:1778
Method
test_tree_all_finite_passes_finite
(self)
chex/_src/asserts_test.py:923
Method
test_tree_all_finite_should_fail_inf
(self)
chex/_src/asserts_test.py:928
Method
test_tree_flatten_with_keys
(self)
chex/_src/dataclass_test.py:372
Method
test_tree_map_with_keys
(self)
chex/_src/dataclass_test.py:387
Method
test_tree_map_with_keys_traversal_order
(self)
chex/_src/dataclass_test.py:407
Method
test_tuple_conversion
(self)
chex/_src/dataclass_test.py:518
Method
test_tuple_rev_conversion
(self, frozen)
chex/_src/dataclass_test.py:536
Method
test_type_should_fail_array
(self, array, dtype, wrong_type)
chex/_src/asserts_test.py:687
Method
test_type_should_fail_mixed
(self)
chex/_src/asserts_test.py:714
Method
test_type_should_fail_scalar
(self, scalars, wrong_type)
chex/_src/asserts_test.py:674
Method
test_type_should_fail_wrong_length
(self, array, wrong_type)
chex/_src/asserts_test.py:736
Method
test_type_should_pass_array
(self, array, dtype, expected_type)
chex/_src/asserts_test.py:710
Method
test_type_should_pass_mixed
(self)
chex/_src/asserts_test.py:724
Method
test_type_should_pass_scalar
(self, array, expected_type)
chex/_src/asserts_test.py:700
Method
test_unexpected_kwargs
(self)
chex/_src/dataclass_test.py:507
Method
test_unfrozen_dataclass_is_mutable
(self)
chex/_src/dataclass_test.py:474
Method
test_uninspected_checks
(self)
chex/_src/asserts_chexify_test.py:189
Method
test_unknown_argument
(self)
chex/_src/variants_test.py:461
Method
test_unsupported_device
(self)
chex/_src/asserts_test.py:1753
Method
test_unused_variant
(self)
chex/_src/variants_test.py:428
Method
test_useful_errors
(self)
chex/_src/variants_test.py:271
Method
test_useful_failure
(self)
chex/_src/variants_test.py:293
Method
test_useful_failures
(self)
chex/_src/variants_test.py:266
Method
test_var_type
(self)
chex/_src/variants_test.py:477
Method
test_var_type_fetch
(self)
chex/_src/variants_test.py:481
Method
test_vmapped_fn_named_params
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:575
Method
test_warn_deprecated_function
(self)
chex/_src/warnings_test.py:63
Method
test_warn_deprecated_function_no_replacement
(self)
chex/_src/warnings_test.py:53
Method
test_warn_only_n_pos_args_in_future
(self)
chex/_src/warnings_test.py:47
Method
test_with_default_params
(self, fake_pmap, fake_jit)
chex/_src/fake_test.py:344
Function
test_with_kwargs
(self)
chex/_src/variants_test.py:128
Method
test_with_kwargs
(self, fake_pmap, fake_jit)
chex/_src/fake_test.py:207
Function
test_with_ndarray_args
(self)
chex/_src/variants_test.py:155
Method
test_with_partial
(self, fake_pmap, fake_jit)
chex/_src/fake_test.py:317
Function
test_with_scalar_args
(self)
chex/_src/variants_test.py:112
Method
test_with_static_broadcasted_argnums
(self, fake_pmap, static_argnums)
chex/_src/fake_test.py:229
Method
test_wrong_base_class
(self)
chex/_src/variants_test.py:316
Method
test_wrong_order_of_wrapping
(self)
chex/_src/asserts_chexify_test.py:297
Function
tree_error_msg_fn
(l_1: _ai.TLeaf, l_2: _ai.TLeaf, path: str, i_1: int, i_2: int)
chex/_src/asserts.py:1426
Method
unraisablehook
(unraisable)
chex/_src/asserts_chexify_test.py:201
Function
variants
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
Method
visit_without_dicts
(x)
chex/_src/dataclass_test.py:339
Method
vmapped_fn
(arg_0, arg_1)
chex/_src/variants_test.py:798
Function
warn_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
Function
warn_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
Function
with_jittable_assertions
An alias for `chexify` (see the docs).
chex/_src/asserts_chexify.py:244
Function
wrapper
(*args, **kwargs)
chex/_src/warnings.py:43
Function
wrapper
(backend, *args, **kwargs)
chex/_src/restrict_backends.py:47
Function
wrapper
(*args, **kwargs)
chex/_src/variants.py:361
← previous
501–578 of 578, ranked by callers