MCPcopy Create free account

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

Functions1,092 in github.com/google-deepmind/penzai

⨍ Functions 1,092◇ Types & classes 360↳ Endpoints 3
Methodtest_saving_all_intermediates
(self, shapes_only, intermediates_group)
tests/toolshed/save_intermediates_test.py:49
Methodtest_scan
(self)
tests/core/named_axes_test.py:1119
Methodtest_select
(self)
tests/core/selectors_test.py:59
Methodtest_select_and_set_by_path
(self)
tests/core/selectors_test.py:776
Methodtest_select_at_accessor__example_1
(self)
tests/core/selectors_test.py:84
Methodtest_select_at_accessor__example_2
(self)
tests/core/selectors_test.py:120
Methodtest_select_at_accessor_can_select_singletons
(self)
tests/core/selectors_test.py:195
Methodtest_select_at_accessor_deprecated_auto_multiple
(self)
tests/core/selectors_test.py:157
Methodtest_select_at_accessor_fails_not_multiple
(self)
tests/core/selectors_test.py:181
Methodtest_select_at_accessor_multiple_and_chaining
(self)
tests/core/selectors_test.py:138
Methodtest_select_at_childless
(self)
tests/core/selectors_test.py:517
Methodtest_select_at_children
(self)
tests/core/selectors_test.py:465
Methodtest_select_at_equal_to__example_1
(self)
tests/core/selectors_test.py:415
Methodtest_select_at_equal_to__example_2
(self)
tests/core/selectors_test.py:429
Methodtest_select_at_instances_of__example_1
(self)
tests/core/selectors_test.py:218
Methodtest_select_at_instances_of__example_2
(self)
tests/core/selectors_test.py:236
Methodtest_select_at_instances_of__example_3
(self)
tests/core/selectors_test.py:256
Methodtest_select_at_keypaths
(self)
tests/core/selectors_test.py:439
Methodtest_select_at_pytree_leaves__example_1
(self)
tests/core/selectors_test.py:481
Methodtest_select_at_pytree_leaves__example_2
(self)
tests/core/selectors_test.py:499
Methodtest_select_at_subtrees_where
(self)
tests/core/selectors_test.py:276
Methodtest_select_at_subtrees_where_innermost
(self)
tests/core/selectors_test.py:395
Methodtest_select_at_subtrees_where_with_absolute_keypath
(self)
tests/core/selectors_test.py:316
Methodtest_select_at_subtrees_where_with_keypath
(self)
tests/core/selectors_test.py:295
Methodtest_select_deselect
(self)
tests/core/selectors_test.py:78
Methodtest_select_selected_selection
(self)
tests/core/selectors_test.py:1066
Methodtest_select_where
(self)
tests/core/selectors_test.py:535
Methodtest_selection_apply
(self)
tests/core/selectors_test.py:794
Methodtest_selection_apply_and_inline
(self)
tests/core/selectors_test.py:1018
Methodtest_selection_apply_keep_selected
(self)
tests/core/selectors_test.py:812
Methodtest_selection_apply_with_keypath
(self)
tests/core/selectors_test.py:831
Methodtest_selection_apply_with_keypath_keep_selected
(self)
tests/core/selectors_test.py:854
Methodtest_selection_apply_with_selected_index
(self)
tests/core/selectors_test.py:883
Methodtest_selection_apply_with_selected_index_keep_selected
(self)
tests/core/selectors_test.py:903
Methodtest_selection_get
(self)
tests/core/selectors_test.py:668
Methodtest_selection_get_by_path
(self)
tests/core/selectors_test.py:684
Methodtest_selection_get_sequence
(self)
tests/core/selectors_test.py:674
Methodtest_selection_insert_after
(self)
tests/core/selectors_test.py:966
Methodtest_selection_insert_before
(self)
tests/core/selectors_test.py:925
Methodtest_selection_remove
(self)
tests/core/selectors_test.py:1007
Methodtest_selection_rendering
(self)
tests/treescope/renderer_test.py:304
Methodtest_selection_set
(self)
tests/core/selectors_test.py:706
Methodtest_selection_set_by_path_dict
(self)
tests/core/selectors_test.py:742
Methodtest_selection_set_by_path_fn
(self)
tests/core/selectors_test.py:760
Methodtest_selection_set_sequence
(self)
tests/core/selectors_test.py:724
Methodtest_selection_size_summaries
(self)
tests/core/selectors_test.py:655
Methodtest_setup_flax
(self)
tests/toolshed/unflaxify_test.py:50
Methodtest_setup_flax
(self)
tests/deprecated/v1/toolshed/unflaxify_test.py:51
Methodtest_shared_parameter_safety_checks
(self)
tests/deprecated/v1/nn/parameters_test.py:415
Methodtest_shared_parameters
(self)
tests/deprecated/v1/nn/parameters_test.py:235
Methodtest_shared_variables_and_states
(self)
tests/toolshed/unflaxify_test.py:87
Methodtest_shared_variables_and_states
(self)
tests/deprecated/v1/toolshed/unflaxify_test.py:89
Methodtest_side_input_constant
(self, keep_unused)
tests/deprecated/v1/data_effects/side_input_test.py:93
Methodtest_side_input_from_input_tuple_nested_pairs
(self)
tests/deprecated/v1/data_effects/side_input_test.py:51
Methodtest_side_input_from_input_tuple_triple
(self)
tests/deprecated/v1/data_effects/side_input_test.py:70
Methodtest_side_output_collect_subset
(self)
tests/deprecated/v1/data_effects/side_output_test.py:73
Methodtest_side_outputs
(self)
tests/deprecated/v1/data_effects/side_output_test.py:50
Methodtest_simple_array
(self)
tests/core/shapecheck_test.py:54
Methodtest_simple_capture
(self)
tests/toolshed/isolate_submodel_test.py:27
Methodtest_simple_capture
(self)
tests/deprecated/v1/toolshed/isolate_submodel_test.py:27
Methodtest_simple_flax
(self)
tests/toolshed/unflaxify_test.py:27
Methodtest_simple_flax
(self)
tests/deprecated/v1/toolshed/unflaxify_test.py:27
Methodtest_simple_named
(self)
tests/core/shapecheck_test.py:94
Methodtest_slice
(self)
tests/core/misc_util_test.py:47
Methodtest_softmax
(self)
tests/nn/basic_ops_test.py:29
Methodtest_softmax
(self)
tests/deprecated/v1/nn/basic_ops_test.py:29
Methodtest_solve_dimension_vars
(self)
tests/core/shapecheck_test.py:138
Methodtest_stack_call
(self)
tests/nn/layer_stack_test.py:46
Methodtest_standardize_multi
(self)
tests/nn/standardization_test.py:31
Methodtest_standardize_multi
(self)
tests/deprecated/v1/nn/standardization_test.py:28
Methodtest_standardize_one
(self)
tests/nn/standardization_test.py:24
Methodtest_standardize_one
(self)
tests/deprecated/v1/nn/standardization_test.py:24
Methodtest_state_sharing_explicit
(self)
tests/deprecated/v1/data_effects/local_state_test.py:265
Methodtest_state_sharing_incoherent
(self, freeze)
tests/deprecated/v1/data_effects/local_state_test.py:228
Methodtest_state_sharing_initial
(self)
tests/deprecated/v1/data_effects/local_state_test.py:196
Methodtest_state_workflow
(self, lazy)
tests/deprecated/v1/data_effects/local_state_test.py:93
Methodtest_struct_attribute_helpers
(self)
tests/core/struct_pytree_dataclass_test.py:357
Methodtest_struct_keypath_custom_keys
(self)
tests/core/struct_pytree_dataclass_test.py:399
Methodtest_struct_pytree_structure
(self)
tests/core/struct_pytree_dataclass_test.py:114
Methodtest_struct_repr_inherited
(self)
tests/core/struct_pytree_dataclass_test.py:392
Methodtest_struct_select
(self)
tests/core/struct_pytree_dataclass_test.py:375
Methodtest_structure_into_pytree
(self)
tests/core/shapecheck_test.py:687
Methodtest_train_deterministic_mlp
(self)
tests/models/simple_mlp_test.py:55
Methodtest_train_deterministic_mlp
(self)
tests/deprecated/v1/example_models/simple_mlp_test.py:45
Methodtest_tree_flatten_exactly_one_level_leaf
(self)
tests/core/misc_util_test.py:41
Methodtest_tree_flatten_exactly_one_level_nested_pytree
(self)
tests/core/misc_util_test.py:26
Methodtest_tree_order_like
(self)
tests/core/named_axes_test.py:853
Methodtest_unbind_variables
(self)
tests/core/variables_test.py:71
Methodtest_undecorated_struct_is_abstract
(self)
tests/core/struct_pytree_dataclass_test.py:344
Methodtest_uninitialized_parameter_checks_dtype
(self)
tests/deprecated/v1/nn/parameters_test.py:125
Methodtest_uninitialized_parameter_checks_shape
(self)
tests/deprecated/v1/nn/parameters_test.py:111
Methodtest_uninitialized_parameter_checks_structure
(self)
tests/deprecated/v1/nn/parameters_test.py:97
Methodtest_uninitialized_parameter_initialization
(self)
tests/deprecated/v1/nn/parameters_test.py:52
Methodtest_untag_then_tag
(self)
tests/core/named_axes_test.py:93
Methodtest_untag_then_tag_prefix
(self)
tests/core/named_axes_test.py:118
Methodtest_unwrap
(self)
tests/core/named_axes_test.py:157
Methodtest_variable_auto_label
(self)
tests/core/variables_test.py:35
Methodtest_variable_auto_label_scoped
(self)
tests/core/variables_test.py:41
Methodtest_variable_basic
(self, var_cls)
tests/core/variables_test.py:27
Methodtest_variable_freeze_unfreeze
(self)
tests/core/variables_test.py:198
← previousnext →901–1,000 of 1,092, ranked by callers