Index _ | A | B | C | D | E | F | G | I | L | M | N | O | P | Q | R | S | T | U | V | W _ __call__() (arraycontext.impl.pytato.compile.BaseLazilyCompilingFunctionCaller method) (arraycontext.impl.pytato.compile.CompiledFunction method) (arraycontext.PytestPyOpenCLArrayContextFactory method) __getitem__ (arraycontext.Array attribute) __init__() (arraycontext.BcastUntilActxArray method) (arraycontext.NumpyArrayContext method) (arraycontext.PytatoPyOpenCLArrayContext method) (arraycontext.PytestPyOpenCLArrayContextFactory method) A allocator (arraycontext.PyOpenCLArrayContext attribute) (arraycontext.PytatoPyOpenCLArrayContext attribute) Allocator (class in cl_array) ArithArrayContainer (in module arraycontext) Array (class in arraycontext) array_types (arraycontext.ArrayContext attribute) ArrayArg (class in arraycontext) ArrayContainer (class in arraycontext) ArrayContainerT (class in arraycontext) (class in arraycontext.container) (class in arraycontext.traversal) (class in arraycontext.typing) arraycontext module ArrayContext (class in arraycontext) arraycontext.container module arraycontext.container.arithmetic module arraycontext.container.dataclass module arraycontext.container.traversal module arraycontext.context module arraycontext.impl.jax module arraycontext.impl.numpy module arraycontext.impl.pyopencl module arraycontext.impl.pyopencl.taggable_cl_array module arraycontext.impl.pytato module arraycontext.impl.pytato.compile module arraycontext.impl.pytato.utils module arraycontext.loopy module arraycontext.metadata module arraycontext.pytest module arraycontext.typing module ArrayOrArithContainer (in module arraycontext) ArrayOrArithContainerOrScalar (in module arraycontext) ArrayOrArithContainerOrScalarT (class in arraycontext) ArrayOrArithContainerT (class in arraycontext) ArrayOrContainer (class in arraycontext.traversal) (in module arraycontext) ArrayOrContainerOrScalar (class in dummy_refs) (in module arraycontext) ArrayOrContainerOrScalarT (class in arraycontext) ArrayOrContainerT (class in arraycontext) ArrayOrNames (class in dummy_refs) ArrayOrNamesTc (class in arraycontext.impl.pytato.utils) ArrayOrScalar (class in arraycontext.traversal) (in module arraycontext) ArrayOrScalarT (class in arraycontext) ArrayT (class in arraycontext) axes (arraycontext.impl.pyopencl.taggable_cl_array.TaggableCLArray attribute) Axis (class in arraycontext.impl.pyopencl.taggable_cl_array) B BaseLazilyCompilingFunctionCaller (class in arraycontext.impl.pytato.compile) BcastUntilActxArray (class in arraycontext) C call_loopy() (arraycontext.ArrayContext method) compile() (arraycontext.ArrayContext method) (arraycontext.PytatoPyOpenCLArrayContext method) CompiledFunction (class in arraycontext.impl.pytato.compile) ContainerOrScalarT (class in arraycontext) context (arraycontext.PyOpenCLArrayContext attribute) D dataclass_array_container() (in module arraycontext) deserialize_container() (in module arraycontext) dtype (arraycontext.Array attribute) E EagerJAXArrayContext (class in arraycontext) einsum() (arraycontext.ArrayContext method) EllipsisType (class in arraycontext) F f (arraycontext.impl.pytato.compile.BaseLazilyCompilingFunctionCaller attribute) flat_size_and_dtype() (in module arraycontext) flatten() (in module arraycontext) freeze() (arraycontext.ArrayContext method) (in module arraycontext) freeze_thaw() (arraycontext.ArrayContext method) from_numpy() (arraycontext.ArrayContext method) FromArrayContextCompile (class in arraycontext.impl.pytato.compile) G GenericAlias (class in arraycontext.container) get_container_context_opt() (in module arraycontext) get_container_context_recursively() (in module arraycontext) get_container_context_recursively_opt() (in module arraycontext) I input_id_to_name_in_program (arraycontext.impl.pytato.compile.CompiledFunction attribute) InstructionBase (class in arraycontext) Integer (class in dummy_refs) is_array_container_type() (in module arraycontext) L LazilyJAXCompilingFunctionCaller (class in arraycontext.impl.pytato.compile) LazilyPyOpenCLCompilingFunctionCaller (class in arraycontext.impl.pytato.compile) M make_loopy_program() (in module arraycontext) map_array_container() (in module arraycontext) map_reduce_array_container() (in module arraycontext) mapped_over_array_containers() (in module arraycontext) module arraycontext arraycontext.container arraycontext.container.arithmetic arraycontext.container.dataclass arraycontext.container.traversal arraycontext.context arraycontext.impl.jax arraycontext.impl.numpy arraycontext.impl.pyopencl arraycontext.impl.pyopencl.taggable_cl_array arraycontext.impl.pytato arraycontext.impl.pytato.compile arraycontext.impl.pytato.utils arraycontext.loopy arraycontext.metadata arraycontext.pytest arraycontext.typing multimap_array_container() (in module arraycontext) multimap_reduce_array_container() (in module arraycontext) multimapped_over_array_containers() (in module arraycontext) N name (arraycontext.metadata.NameHint attribute) NameHint (class in arraycontext.metadata) ndarray (class in np) NotAnArrayContainerError np (arraycontext.ArrayContext attribute) NumpyArrayContext (class in arraycontext) O outer() (in module arraycontext) outline() (arraycontext.ArrayContext method) P P (class in arraycontext) PyOpenCLArrayContext (class in arraycontext) pytato_program (arraycontext.impl.pytato.compile.CompiledFunction attribute) PytatoJAXArrayContext (class in arraycontext) PytatoPyOpenCLArrayContext (class in arraycontext) pytest_generate_tests_for_array_contexts() (in module arraycontext) PytestArrayContextFactory (class in arraycontext) PytestPyOpenCLArrayContextFactory (class in arraycontext) Q queue (arraycontext.PyOpenCLArrayContext attribute) (arraycontext.PytatoPyOpenCLArrayContext attribute) R rec_map_array_container() (in module arraycontext) rec_map_container() (in module arraycontext) rec_map_reduce_array_container() (in module arraycontext) rec_multimap_array_container() (in module arraycontext) rec_multimap_reduce_array_container() (in module arraycontext) register_multivector_as_array_container() (in module arraycontext) S ScalarLike (class in dummy_refs) (in module arraycontext) SerializationKey (class in arraycontext.container) (in module arraycontext) serialize_container() (in module arraycontext) SerializedContainer (class in arraycontext.container) (in module arraycontext) shape (arraycontext.Array attribute) size (arraycontext.Array attribute) stringify_array_container_tree() (in module arraycontext) SubstitutionRule (class in arraycontext) T T (class in arraycontext.container.dataclass) tabulate_profiling_data() (in module arraycontext.impl.pytato.utils) tag() (arraycontext.ArrayContext method) tag_axes() (in module arraycontext) tag_axis() (arraycontext.ArrayContext method) TaggableCLArray (class in arraycontext.impl.pyopencl.taggable_cl_array) tags (arraycontext.impl.pyopencl.taggable_cl_array.TaggableCLArray attribute) TemporaryVariable (class in arraycontext) thaw() (arraycontext.ArrayContext method) (in module arraycontext) to_numpy() (arraycontext.ArrayContext method) to_tagged_cl_array() (in module arraycontext.impl.pyopencl.taggable_cl_array) ToTagSetConvertible (class in dummy_refs) transfer_from_numpy() (in module arraycontext.impl.pytato.utils) transfer_to_numpy() (in module arraycontext.impl.pytato.utils) transform_dag() (arraycontext.PytatoJAXArrayContext method) (arraycontext.PytatoPyOpenCLArrayContext method) transform_loopy_program() (arraycontext.PyOpenCLArrayContext method) TypeT (class in arraycontext) U unflatten() (in module arraycontext) UnionType (class in arraycontext.container) V ValueArg (class in arraycontext) W with_container_arithmetic() (in module arraycontext)