Index

_

__bool__() (check_shapes.config.ShapeCheckingState method) (check_shapes.ShapeCheckingState method) __eq__() (check_shapes.error_contexts.ArgumentContext method) (check_shapes.error_contexts.AttributeContext method) (check_shapes.error_contexts.IndexContext method) (check_shapes.error_contexts.MappingKeyContext method) (check_shapes.error_contexts.MappingValueContext method) (check_shapes.error_contexts.ObjectTypeContext method) (check_shapes.error_contexts.ObjectValueContext method) __hash__() (check_shapes.error_contexts.ArgumentContext method) (check_shapes.error_contexts.AttributeContext method) (check_shapes.error_contexts.IndexContext method) (check_shapes.error_contexts.LarkUnexpectedInputContext method) (check_shapes.error_contexts.MappingKeyContext method) (check_shapes.error_contexts.MappingValueContext method) (check_shapes.error_contexts.ObjectTypeContext method) (check_shapes.error_contexts.ObjectValueContext method) __repr__() (check_shapes.argument_ref.AllElementsRef method) (check_shapes.argument_ref.ArgumentRef method) (check_shapes.argument_ref.AttributeArgumentRef method) (check_shapes.argument_ref.IndexArgumentRef method) (check_shapes.argument_ref.KeysRef method) (check_shapes.argument_ref.RootArgumentRef method) (check_shapes.argument_ref.ValuesRef method) (check_shapes.bool_specs.ParsedAndBoolSpec method) (check_shapes.bool_specs.ParsedArgumentRefBoolSpec method) (check_shapes.bool_specs.ParsedNotBoolSpec method) (check_shapes.bool_specs.ParsedOrBoolSpec method)

_assert() (check_shapes.checker.ShapeChecker method) (check_shapes.ShapeChecker method) _CachedParser (class in check_shapes.parser) _check_dim() (check_shapes.checker.ShapeChecker method) (check_shapes.ShapeChecker method) _guess_indent() (check_shapes.parser._RewritedocString method) _InheritCheckShapes (class in check_shapes.inheritance) _match_dims() (check_shapes.checker.ShapeChecker method) (check_shapes.ShapeChecker method) _NO_VALUE (in module check_shapes.error_contexts) _ObservedDim (class in check_shapes.checker) _ObservedDims (class in check_shapes.checker) _paren_repr() (in module check_shapes.bool_specs) _parse_checks() (check_shapes.checker.ShapeChecker method) (check_shapes.ShapeChecker method) _ParseSpec (class in check_shapes.parser) _RewritedocString (class in check_shapes.parser) _ShapeCheck (class in check_shapes.checker) _token_children() (in module check_shapes.parser) _tree_children() (in module check_shapes.parser) _TreeVisitor (class in check_shapes.parser) _VariableState (class in check_shapes.checker)

A

actual (check_shapes.checker._ShapeCheck attribute) add_columned_line() (check_shapes.error_contexts.MessageBuilder method) add_context() (check_shapes.checker.ShapeChecker method) (check_shapes.ShapeChecker method) add_is_compiled_mode() (in module check_shapes.config) add_line() (check_shapes.error_contexts.MessageBuilder method)

add_wrapper_post_processor() (in module check_shapes.decorator) AllElementsRef (class in check_shapes.argument_ref) ArgumentContext (class in check_shapes.error_contexts) ArgumentRef (class in check_shapes.argument_ref) ArgumentReferenceError AttributeArgumentRef (class in check_shapes.argument_ref) AttributeContext (class in check_shapes.error_contexts)

B

BoolTest (class in check_shapes.bool_specs)

build() (check_shapes.error_contexts.MessageBuilder method)

C

check_and_update() (check_shapes.checker._ObservedDim method) (check_shapes.checker._ObservedDims method) check_shape() (check_shapes.checker.ShapeChecker method) (check_shapes.ShapeChecker method) (in module check_shapes) (in module check_shapes.checker_context) check_shapes module check_shapes() (check_shapes.checker.ShapeChecker method) (check_shapes.ShapeChecker method) (in module check_shapes) (in module check_shapes.decorator) check_shapes.accessors module check_shapes.argument_ref module check_shapes.base_types module check_shapes.bool_specs module check_shapes.checker module check_shapes.checker_context module check_shapes.config module

check_shapes.decorator module check_shapes.error_contexts module check_shapes.exceptions module check_shapes.inheritance module check_shapes.integration module check_shapes.integration.tf module check_shapes.integration.tfp module check_shapes.integration.torch module check_shapes.parser module check_shapes.shapes module check_shapes.specs module CheckShapesError ConditionContext (class in check_shapes.error_contexts)

D

DelegatingArgumentRef (class in check_shapes.argument_ref) Dimension (in module check_shapes) (in module check_shapes.base_types) disable_check_shapes() (in module check_shapes) (in module check_shapes.config)

DISABLED (check_shapes.config.ShapeCheckingState attribute) (check_shapes.ShapeCheckingState attribute) DocstringFormat (class in check_shapes) (class in check_shapes.config) DocstringParseError

E

EAGER_MODE_ONLY (check_shapes.config.ShapeCheckingState attribute) (check_shapes.ShapeCheckingState attribute) ENABLED (check_shapes.config.ShapeCheckingState attribute) (check_shapes.ShapeCheckingState attribute)

ErrorContext (class in check_shapes) (class in check_shapes.error_contexts) expected (check_shapes.checker._ShapeCheck attribute)

F

finished (check_shapes.checker._ShapeCheck property)

FunctionCallContext (class in check_shapes.error_contexts) FunctionDefinitionContext (class in check_shapes.error_contexts)

G

get() (check_shapes.argument_ref.ArgumentRef method) (check_shapes.argument_ref.DelegatingArgumentRef method) (check_shapes.argument_ref.RootArgumentRef method) (check_shapes.bool_specs.ParsedAndBoolSpec method) (check_shapes.bool_specs.ParsedArgumentRefBoolSpec method) (check_shapes.bool_specs.ParsedBoolSpec method) (check_shapes.bool_specs.ParsedNotBoolSpec method) (check_shapes.bool_specs.ParsedOrBoolSpec method) get_check_shapes() (in module check_shapes) (in module check_shapes.accessors) get_drop_frames() (in module check_shapes) (in module check_shapes.config)

get_enable_check_shapes() (in module check_shapes) (in module check_shapes.config) get_enable_function_call_precompute() (in module check_shapes) (in module check_shapes.config) get_rewrite_docstrings() (in module check_shapes) (in module check_shapes.config) get_shape() (in module check_shapes) (in module check_shapes.shapes) get_shape_checker() (in module check_shapes) (in module check_shapes.checker_context)

I

indent() (check_shapes.error_contexts.MessageBuilder method) IndexArgumentRef (class in check_shapes.argument_ref) IndexContext (class in check_shapes.error_contexts) inherit_check_shapes() (in module check_shapes) (in module check_shapes.inheritance)

install_tf_integration() (in module check_shapes.integration.tf) install_tfp_integration() (in module check_shapes.integration.tfp) install_torch_integration() (in module check_shapes.integration.torch) is_result (check_shapes.argument_ref.ArgumentRef property) (check_shapes.argument_ref.DelegatingArgumentRef property) (check_shapes.argument_ref.RootArgumentRef property)

K

KeysRef (class in check_shapes.argument_ref)

L

LarkUnexpectedInputContext (class in check_shapes.error_contexts)

M

map_context() (check_shapes.argument_ref.AttributeArgumentRef method) (check_shapes.argument_ref.DelegatingArgumentRef method) (check_shapes.argument_ref.IndexArgumentRef method) map_value() (check_shapes.argument_ref.AllElementsRef method) (check_shapes.argument_ref.AttributeArgumentRef method) (check_shapes.argument_ref.DelegatingArgumentRef method) (check_shapes.argument_ref.IndexArgumentRef method) (check_shapes.argument_ref.KeysRef method) (check_shapes.argument_ref.ValuesRef method) MappingKeyContext (class in check_shapes.error_contexts) MappingValueContext (class in check_shapes.error_contexts) maybe_get_check_shapes() (in module check_shapes.accessors) MessageBuilder (class in check_shapes.error_contexts) module check_shapes check_shapes.accessors check_shapes.argument_ref check_shapes.base_types check_shapes.bool_specs check_shapes.checker check_shapes.checker_context check_shapes.config check_shapes.decorator check_shapes.error_contexts check_shapes.exceptions check_shapes.inheritance check_shapes.integration check_shapes.integration.tf check_shapes.integration.tfp check_shapes.integration.torch check_shapes.parser check_shapes.shapes check_shapes.specs

MultipleElementBoolContext (class in check_shapes.error_contexts)

N

NONE (check_shapes.config.DocstringFormat attribute) (check_shapes.DocstringFormat attribute)

NoShapeError NoteContext (class in check_shapes.error_contexts) null_check_shapes() (in module check_shapes.decorator)

O

ObjectTypeContext (class in check_shapes.error_contexts) ObjectValueContext (class in check_shapes.error_contexts)

observed_dim (check_shapes.checker._VariableState attribute) observed_dims (check_shapes.checker._VariableState attribute) on_wrap() (check_shapes.decorator.WrapperPostProcessor method)

P

ParallelContext (class in check_shapes.error_contexts) parse_and_rewrite_docstring() (in module check_shapes.parser) parse_function_spec() (in module check_shapes.parser) parse_tensor_spec() (in module check_shapes.parser) ParsedAndBoolSpec (class in check_shapes.bool_specs) ParsedArgumentRefBoolSpec (class in check_shapes.bool_specs) ParsedBoolSpec (class in check_shapes.bool_specs) ParsedNotBoolSpec (class in check_shapes.bool_specs) ParsedOrBoolSpec (class in check_shapes.bool_specs) ParserInputContext (class in check_shapes.error_contexts) precompute() (check_shapes.error_contexts.FunctionCallContext method) print() (check_shapes.error_contexts.ArgumentContext method) (check_shapes.error_contexts.AttributeContext method) (check_shapes.error_contexts.ConditionContext method) (check_shapes.error_contexts.ErrorContext method) (check_shapes.error_contexts.FunctionCallContext method) (check_shapes.error_contexts.FunctionDefinitionContext method) (check_shapes.error_contexts.IndexContext method) (check_shapes.error_contexts.LarkUnexpectedInputContext method) (check_shapes.error_contexts.MappingKeyContext method) (check_shapes.error_contexts.MappingValueContext method) (check_shapes.error_contexts.MultipleElementBoolContext method) (check_shapes.error_contexts.NoteContext method) (check_shapes.error_contexts.ObjectTypeContext method) (check_shapes.error_contexts.ObjectValueContext method) (check_shapes.error_contexts.ParallelContext method) (check_shapes.error_contexts.ShapeContext method) (check_shapes.error_contexts.StackContext method) (check_shapes.error_contexts.TensorSpecContext method) (check_shapes.error_contexts.VariableContext method) (check_shapes.ErrorContext method)

R

register_get_shape() (in module check_shapes) (in module check_shapes.shapes) root_argument_name (check_shapes.argument_ref.ArgumentRef property) (check_shapes.argument_ref.DelegatingArgumentRef property) (check_shapes.argument_ref.RootArgumentRef property)

RootArgumentRef (class in check_shapes.argument_ref)

S

set_check_shapes() (in module check_shapes.accessors) set_drop_frames() (in module check_shapes) (in module check_shapes.config) set_enable_check_shapes() (in module check_shapes) (in module check_shapes.config) set_enable_function_call_precompute() (in module check_shapes) (in module check_shapes.config) set_rewrite_docstrings() (in module check_shapes) (in module check_shapes.config) set_shape_checker() (in module check_shapes.checker_context) Shape (in module check_shapes) (in module check_shapes.base_types)

ShapeChecker (class in check_shapes) (class in check_shapes.checker) ShapeCheckingState (class in check_shapes) (class in check_shapes.config) ShapeContext (class in check_shapes.error_contexts) ShapeMismatchError SpecificationParseError SPHINX (check_shapes.config.DocstringFormat attribute) (check_shapes.DocstringFormat attribute) StackContext (class in check_shapes.error_contexts)

T

TensorSpecContext (class in check_shapes.error_contexts)

U

uses (check_shapes.checker._VariableState attribute)

V

ValuesRef (class in check_shapes.argument_ref)

VariableContext (class in check_shapes.error_contexts) VariableTypeError

W

waiting_for_varrank (check_shapes.checker._VariableState attribute)

WrapperPostProcessor (class in check_shapes.decorator)