__init__.py __pycache__/ assume_constant_result.py autograd_function.py class_method.py cond_branch_class_method.py cond_branch_nested_function.py cond_branch_nonlocal_variables.py cond_closed_over_variable.py cond_operands.py cond_predicate.py constrain_as_size_example.py constrain_as_value_example.py decorator.py dictionary.py dynamic_shape_assert.py dynamic_shape_constructor.py dynamic_shape_if_guard.py dynamic_shape_map.py dynamic_shape_round.py dynamic_shape_slicing.py dynamic_shape_view.py fn_with_kwargs.py list_contains.py list_unpack.py model_attr_mutation.py nested_function.py null_context_manager.py optional_input.py pytree_flatten.py scalar_output.py specialized_attribute.py static_for_loop.py static_if.py tensor_setattr.py type_reflection_method.py unsupported_operator.py user_input_mutation.py