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
MethodtestTraverse
(self, test_type)
chex/_src/dataclass_test.py:329
Methodtest_1
(self)
chex/_src/variants_test.py:508
Methodtest_2
(self)
chex/_src/variants_test.py:512
Methodtest_3
(self)
chex/_src/variants_test.py:516
Methodtest_4
(self)
chex/_src/variants_test.py:523
Methodtest_all_variants
(self)
chex/_src/variants_test.py:547
Methodtest_all_variants_no_parens
(self)
chex/_src/variants_test.py:553
Methodtest_api
(self)
chex/_src/asserts_internal_test.py:108
Methodtest_arg
(self)
chex/_src/variants_test.py:458
Methodtest_assert
(self, init_type, kwargs, static_arg, max_traces)
chex/_src/asserts_test.py:151
Methodtest_assert_axis_dimension_axis_invalid
(self)
chex/_src/asserts_test.py:759
Methodtest_assert_axis_dimension_fail
(self)
chex/_src/asserts_test.py:751
Methodtest_assert_axis_dimension_gt_axis_invalid
(self)
chex/_src/asserts_test.py:781
Methodtest_assert_axis_dimension_gt_fail
(self)
chex/_src/asserts_test.py:772
Methodtest_assert_axis_dimension_gt_pass
(self)
chex/_src/asserts_test.py:766
Methodtest_assert_axis_dimension_gteq_axis_invalid
(self)
chex/_src/asserts_test.py:803
Methodtest_assert_axis_dimension_gteq_fail
(self)
chex/_src/asserts_test.py:794
Methodtest_assert_axis_dimension_gteq_pass
(self)
chex/_src/asserts_test.py:788
Methodtest_assert_axis_dimension_lt_axis_invalid
(self)
chex/_src/asserts_test.py:825
Methodtest_assert_axis_dimension_lt_fail
(self)
chex/_src/asserts_test.py:816
Methodtest_assert_axis_dimension_lt_pass
(self)
chex/_src/asserts_test.py:810
Methodtest_assert_axis_dimension_lteq_axis_invalid
(self)
chex/_src/asserts_test.py:847
Methodtest_assert_axis_dimension_lteq_fail
(self)
chex/_src/asserts_test.py:838
Methodtest_assert_axis_dimension_lteq_pass
(self)
chex/_src/asserts_test.py:832
Methodtest_assert_axis_dimension_pass
(self)
chex/_src/asserts_test.py:745
Methodtest_assert_axis_dimension_string_tensor
(self)
chex/_src/asserts_test.py:854
Methodtest_assert_equal_fail
(self, first, second)
chex/_src/asserts_test.py:1889
Methodtest_assert_equal_pass
(self, first, second)
chex/_src/asserts_test.py:1870
Methodtest_assert_equal_pass_on_arrays
(self)
chex/_src/asserts_test.py:1873
Methodtest_assert_is_divisible
(self)
chex/_src/asserts_test.py:1896
Methodtest_assert_is_divisible_fail
(self)
chex/_src/asserts_test.py:1899
Methodtest_assert_jitted
(self)
chex/_src/fake_test.py:101
Methodtest_assert_pmapped
(self)
chex/_src/fake_test.py:90
Methodtest_assert_rank_should_fail_sequence
(self, arrays, ranks)
chex/_src/asserts_test.py:617
Methodtest_assert_tree_all_finite
(self)
chex/_src/asserts_chexify_test.py:589
Methodtest_assert_tree_has_only_ndarrays
(self)
chex/_src/asserts_test.py:1186
Methodtest_assert_tree_is_on_device
(self)
chex/_src/asserts_test.py:1271
Methodtest_assert_tree_is_on_host
(self)
chex/_src/asserts_test.py:1200
Methodtest_assert_tree_is_sharded
(self)
chex/_src/asserts_test.py:1357
Methodtest_assert_tree_no_nones
(self)
chex/_src/asserts_test.py:904
Methodtest_assert_tree_shape_mismatched_should_fail
(self, shape)
chex/_src/asserts_test.py:1567
Methodtest_assert_tree_shape_mismatched_should_pass
(self, shape)
chex/_src/asserts_test.py:1554
Methodtest_assert_tree_shape_prefix
(self, shape)
chex/_src/asserts_test.py:1579
Methodtest_assert_tree_shape_should_fail
(self, shape)
chex/_src/asserts_test.py:1544
Methodtest_assert_tree_shape_should_pass
(self, shape)
chex/_src/asserts_test.py:1534
Methodtest_assert_tree_shape_suffix_long_suffix
(self)
chex/_src/asserts_test.py:1634
Methodtest_assert_tree_shape_suffix_matching
(self, shape)
chex/_src/asserts_test.py:1602
Methodtest_assert_tree_shape_suffix_mismatch
(self, leaf, shape_true, shape)
chex/_src/asserts_test.py:1613
Methodtest_assert_trees_all_close
(self)
chex/_src/asserts_chexify_test.py:644
Methodtest_assert_trees_all_close_bfloat16
(self)
chex/_src/asserts_test.py:1037
Methodtest_assert_trees_all_close_fails_different_structure
(self)
chex/_src/asserts_test.py:1450
Methodtest_assert_trees_all_close_fails_values_differ
(self)
chex/_src/asserts_test.py:1453
Methodtest_assert_trees_all_close_passes_same_tree
(self)
chex/_src/asserts_test.py:997
Methodtest_assert_trees_all_close_passes_values_close_but_not_equal
(self)
chex/_src/asserts_test.py:1012
Methodtest_assert_trees_all_close_passes_values_equal
(self)
chex/_src/asserts_test.py:1006
Methodtest_assert_trees_all_close_strict_mode
(self)
chex/_src/asserts_test.py:1019
Methodtest_assert_trees_all_close_ulp_bfloat16_fails_values_gt_maxulp_apart
( self, )
chex/_src/asserts_test.py:1161
Methodtest_assert_trees_all_close_ulp_bfloat16_passes_values_maxulp_apart
(self)
chex/_src/asserts_test.py:1141
Methodtest_assert_trees_all_close_ulp_bfloat16_passes_values_within_maxulp
( self, )
chex/_src/asserts_test.py:1117
Methodtest_assert_trees_all_close_ulp_fails_values_gt_maxulp_apart
(self)
chex/_src/asserts_test.py:1104
Methodtest_assert_trees_all_close_ulp_jittable_raises_valueerror
(self)
chex/_src/asserts_test.py:1060
Methodtest_assert_trees_all_close_ulp_passes_same_tree
(self)
chex/_src/asserts_test.py:1067
Methodtest_assert_trees_all_close_ulp_passes_values_equal
(self)
chex/_src/asserts_test.py:1074
Methodtest_assert_trees_all_close_ulp_passes_values_maxulp_apart
(self)
chex/_src/asserts_test.py:1093
Methodtest_assert_trees_all_close_ulp_passes_values_within_maxulp
(self)
chex/_src/asserts_test.py:1082
Methodtest_assert_trees_all_equal
(self)
chex/_src/asserts_chexify_test.py:606
Methodtest_assert_trees_all_equal_dtypes
(self)
chex/_src/asserts_test.py:1647
Methodtest_assert_trees_all_equal_fail_values_close_but_not_equal
(self)
chex/_src/asserts_test.py:964
Methodtest_assert_trees_all_equal_none
(self)
chex/_src/asserts_test.py:1712
Methodtest_assert_trees_all_equal_passes_same_tree
(self)
chex/_src/asserts_test.py:949
Methodtest_assert_trees_all_equal_passes_values_equal
(self)
chex/_src/asserts_test.py:958
Methodtest_assert_trees_all_equal_prng_keys
(self)
chex/_src/asserts_test.py:940
Methodtest_assert_trees_all_equal_shapes
(self)
chex/_src/asserts_test.py:1490
Methodtest_assert_trees_all_equal_shapes_and_dtypes
(self)
chex/_src/asserts_test.py:1678
Methodtest_assert_trees_all_equal_sizes
(self)
chex/_src/asserts_test.py:1466
Methodtest_assert_trees_all_equal_strict_mode
(self)
chex/_src/asserts_test.py:973
Methodtest_assert_trees_all_equal_structs
(self)
chex/_src/asserts_test.py:1514
Methodtest_assert_trees_all_equal_with_prng_keys
(self)
chex/_src/asserts_chexify_test.py:629
Methodtest_assert_trees_all_equal_wrong_usage
(self)
chex/_src/asserts_test.py:1701
Methodtest_async_mode
(self)
chex/_src/asserts_chexify_test.py:152
Functiontest_called_variant
(self)
chex/_src/variants_test.py:120
Methodtest_checkify_errors
(self)
chex/_src/asserts_chexify_test.py:236
Methodtest_consistency
(self)
chex/_src/variants_test.py:489
Methodtest_counters
(self)
chex/_src/variants_test.py:526
Methodtest_cpu_assert
(self)
chex/_src/asserts_test.py:1774
Methodtest_create_deprecated_function_alias
(self)
chex/_src/warnings_test.py:72
Methodtest_custom_message
(self)
chex/_src/asserts_chexify_test.py:669
Methodtest_dataclass_instance_fields
(self, dcls)
chex/_src/dataclass_test.py:696
Methodtest_dataclass_mappable_dataclass_false
(self)
chex/_src/dataclass_test.py:444
Methodtest_dataclass_replace
(self, frozen)
chex/_src/dataclass_test.py:419
Methodtest_dataclass_requires_kwargs_by_default
(self)
chex/_src/dataclass_test.py:430
Methodtest_dataclass_tree_leaves
(self, frozen)
chex/_src/dataclass_test.py:360
Methodtest_dataclass_tree_map
(self, frozen)
chex/_src/dataclass_test.py:365
Methodtest_decorator_alias
(self)
chex/_src/dataclass_test.py:646
Methodtest_detects_explicitly_forbidden_cpu_computation
(self)
chex/_src/restrict_backends_test.py:73
Methodtest_detects_explicitly_forbidden_tpu_computation
(self)
chex/_src/restrict_backends_test.py:57
Methodtest_detects_implicitly_forbidden_cpu_computation
(self)
chex/_src/restrict_backends_test.py:65
Methodtest_detects_implicitly_forbidden_tpu_computation
(self)
chex/_src/restrict_backends_test.py:49
Methodtest_different_jit_kwargs
(self, arg_0, arg_1, expected)
chex/_src/variants_test.py:602
Methodtest_disallowed_fields
(self)
chex/_src/dataclass_test.py:588
← previousnext →301–400 of 578, ranked by callers