Index A | B | C | D | E | F | G | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W A abi_name (mlx_lattice.artifact.plan.PlanArgument attribute) (mlx_lattice.artifact.plan.PlanOutput attribute) (mlx_lattice.artifact.PlanArgument attribute) (mlx_lattice.artifact.PlanOutput attribute) activation() (in module lattice_contract.dialect) (in module mlx_lattice.ops.feature) ActivationAttr (class in lattice_contract.dialect) active_count (mlx_lattice.core.coords.alignment.SparseAlignment property) (mlx_lattice.core.coords.occupancy.OccupancyExpansion property) (mlx_lattice.core.coords.occupancy.SparseOccupancy property) (mlx_lattice.core.coords.quantization.SparseQuantization property) (mlx_lattice.core.coords.set_ops.CoordinateSet property) (mlx_lattice.core.tensor.SparseTensor property) active_rows (mlx_lattice.core.coords.alignment.SparseAlignment attribute) (mlx_lattice.core.coords.occupancy.OccupancyExpansion attribute) (mlx_lattice.core.coords.occupancy.SparseOccupancy attribute) (mlx_lattice.core.coords.quantization.SparseQuantization attribute) (mlx_lattice.core.coords.set_ops.CoordinateSet attribute) (mlx_lattice.core.tensor.SparseTensor attribute) (mlx_lattice.core.tensor.SparseTensorComponents attribute) active_rows() (mlx_lattice.core.coords.manager.CoordinateManager method) add_attr() (lattice_contract.schema.DialectSchema method) add_op() (lattice_contract.schema.DialectSchema method) add_type() (lattice_contract.schema.DialectSchema method) align_sparse() (in module mlx_lattice.ops.tensor) args (mlx_lattice.artifact.plan.RuntimePlan attribute) (mlx_lattice.artifact.RuntimePlan attribute) argument() (lattice_contract.mlir.builder.MLIRModuleBuilder method) assembly (lattice_contract.schema.OpDef attribute) astype() (mlx_lattice.core.tensor.SparseTensor method) attr() (lattice_contract.schema.DialectSchema method) attr_param() (in module lattice_contract.schema) AttrDef (class in lattice_contract.schema) attributes (lattice_contract.schema.OpDef attribute) AttrParameter (class in lattice_contract.schema) attrs (lattice_contract.schema.DialectSchema attribute) (mlx_lattice.artifact.plan.PlanOperation attribute) (mlx_lattice.artifact.PlanOperation attribute) avg_pool3d() (in module mlx_lattice.ops.pool) AvgPool3d (class in mlx_lattice.nn.pool) B backend_info() (in module mlx_lattice) batch_counts (mlx_lattice.core.tensor.SparseTensor attribute) (mlx_lattice.core.tensor.SparseTensorComponents attribute) batch_indices (mlx_lattice.core.tensor.SparseTensor property) batch_norm() (in module lattice_contract.dialect) (in module mlx_lattice.ops.feature) batch_norm_features() (in module mlx_lattice.ops.feature) batch_rows (mlx_lattice.core.tensor.SparseTensor property) BatchNorm (class in mlx_lattice.nn.feature) biases (mlx_lattice.core.quantized.QuantizedWeight attribute) BinaryOpAttr (class in lattice_contract.dialect) bits (mlx_lattice.core.quantized.QuantizedWeight attribute) build_generative_relation() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) build_kernel_relation() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) build_knn_relation() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) build_point_voxel_map() (in module mlx_lattice.core.coords.quantization) (in module mlx_lattice.ops.quantization) build_radius_relation() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) build_relation_implicit_gemm_view() (in module mlx_lattice.core.coords.builders) build_sparse_alignment() (in module mlx_lattice.core.coords.alignment) build_submanifold_kernel_relation() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) build_target_kernel_relation() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) build_target_transposed_implicit_gemm_view() (in module mlx_lattice.core.coords.builders) build_transposed_kernel_relation() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) C call_with_trusted_sparse_inputs() (mlx_lattice.artifact.LatticeProgram method) (mlx_lattice.artifact.runtime.LatticeProgram method) canonical_schema() (in module lattice_contract.schema) capacity (mlx_lattice.core.coords.alignment.SparseAlignment property) (mlx_lattice.core.coords.occupancy.OccupancyExpansion property) (mlx_lattice.core.coords.occupancy.SparseOccupancy property) (mlx_lattice.core.coords.quantization.SparseQuantization property) (mlx_lattice.core.coords.set_ops.CoordinateSet property) (mlx_lattice.core.relations.views.NeighborEdges property) (mlx_lattice.core.relations.views.RelationEdges property) (mlx_lattice.core.tensor.SparseTensor property) cat() (in module mlx_lattice.ops.tensor) channels (mlx_lattice.core.tensor.SparseTensor property) child_coords_from_indices() (in module mlx_lattice.core.coords.occupancy) child_indices (mlx_lattice.core.coords.occupancy.OccupancyExpansion attribute) compile_lattice_artifact() (in module mlx_lattice.artifact) (in module mlx_lattice.artifact.runtime) contains_coords() (in module mlx_lattice.core.coords.set_ops) contract (mlx_lattice.core.relations.views.KernelRelation attribute) Conv3d (class in mlx_lattice.nn.conv) conv3d() (in module lattice_contract.dialect) (in module mlx_lattice.ops.conv) conv_transpose3d() (in module lattice_contract.dialect) (in module mlx_lattice.ops.conv) ConvTranspose3d (class in mlx_lattice.nn.conv) coord (lattice_contract.mlir.builder.SparseTensorType attribute) coord_key (mlx_lattice.core.tensor.SparseTensor attribute) coord_manager (mlx_lattice.core.tensor.SparseTensor attribute) CoordAttr (class in lattice_contract.dialect) CoordinateManager (class in mlx_lattice.core.coords.manager) CoordinateMapKey (class in mlx_lattice.core.coords.manager) CoordinateOrdering (class in mlx_lattice.core.coords.ordering) CoordinateSet (class in mlx_lattice.core.coords.set_ops) coords (mlx_lattice.core.coords.alignment.SparseAlignment attribute) (mlx_lattice.core.coords.occupancy.OccupancyExpansion attribute) (mlx_lattice.core.coords.occupancy.SparseOccupancy attribute) (mlx_lattice.core.coords.ordering.CoordinateOrdering attribute) (mlx_lattice.core.coords.quantization.SparseQuantization attribute) (mlx_lattice.core.coords.set_ops.CoordinateSet attribute) (mlx_lattice.core.tensor.SparseTensor attribute) (mlx_lattice.core.tensor.SparseTensorComponents attribute) coords() (mlx_lattice.core.coords.manager.CoordinateManager method) counts (mlx_lattice.core.coords.quantization.SparseQuantization attribute) (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.NeighborRelation attribute) (mlx_lattice.core.relations.views.SparseRelationContract attribute) crop() (in module mlx_lattice.ops.tensor) D decomposed_coordinates (mlx_lattice.core.tensor.SparseTensor property) decomposed_coordinates_and_features (mlx_lattice.core.tensor.SparseTensor property) decomposed_features (mlx_lattice.core.tensor.SparseTensor property) default (lattice_contract.mlir.builder.EmitRegistry attribute) definition (mlx_lattice.artifact.plan.PlanOperation attribute) (mlx_lattice.artifact.PlanOperation attribute) dense_packing() (in module lattice_contract.mlir.builder) dequantize_weight() (in module mlx_lattice.core.quantized) devoxelize() (in module lattice_contract.dialect) (in module mlx_lattice.ops.quantization) diagnostics (mlx_lattice.artifact.LatticeMLIRStatus attribute) (mlx_lattice.artifact.validation.LatticeMLIRStatus attribute) DialectSchema (class in lattice_contract.schema) dilation (mlx_lattice.core.relations.specs.KernelSpec attribute) distances (mlx_lattice.core.relations.views.NeighborRelation attribute) downsample_coords() (in module mlx_lattice.core.coords.set_ops) Dropout (class in mlx_lattice.nn.feature) dropout() (in module mlx_lattice.ops.feature) dtype (lattice_contract.mlir.builder.SparseTensorType attribute) (lattice_contract.mlir.builder.WeightType attribute) (mlx_lattice.core.tensor.SparseTensor property) DuplicateReduction (in module mlx_lattice.ops.tensor) E edge_capacity (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.NeighborRelation property) edge_count (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.NeighborRelation property) (mlx_lattice.core.relations.views.SparseRelationContract property) edge_ids (mlx_lattice.core.relations.views.RelationCSRView attribute) edges (mlx_lattice.core.relations.views.KernelRelation attribute) (mlx_lattice.core.relations.views.NeighborRelation attribute) elementwise() (in module lattice_contract.dialect) embedding_lookup() (in module lattice_contract.dialect) emit() (lattice_contract.mlir.builder.EmitRegistry method) (lattice_contract.mlir.builder.MLIRModuleBuilder method) EmitFn (class in lattice_contract.mlir.builder) EmitRegistry (class in lattice_contract.mlir.builder) EmitResult (in module lattice_contract.mlir.builder) export_components() (mlx_lattice.core.tensor.SparseTensor method) F family (lattice_contract.mlir.builder.WeightType attribute) feats (mlx_lattice.core.tensor.SparseTensor attribute) (mlx_lattice.core.tensor.SparseTensorComponents attribute) feature (lattice_contract.mlir.builder.SparseTensorType attribute) FeatureLayoutAttr (class in lattice_contract.dialect) from_artifact() (mlx_lattice.artifact.LatticeProgram class method) (mlx_lattice.artifact.runtime.LatticeProgram class method) from_components() (mlx_lattice.core.tensor.SparseTensor class method) from_conv() (mlx_lattice.nn.quantized_conv.QuantizedConv3d class method) (mlx_lattice.nn.quantized_conv.QuantizedConvTranspose3d class method) (mlx_lattice.nn.quantized_conv.QuantizedGenerativeConvTranspose3d class method) (mlx_lattice.nn.quantized_conv.QuantizedSubmConv3d class method) from_linear() (mlx_lattice.nn.quantized_feature.QuantizedLinear class method) from_native() (mlx_lattice.artifact.plan.RuntimePlan class method) (mlx_lattice.artifact.RuntimePlan class method) functional_emitter() (in module lattice_contract.mlir.builder) functions (lattice_contract.mlir.builder.EmitRegistry attribute) G gather_aligned_features() (in module mlx_lattice.ops.tensor) gather_neighbor_features() (in module mlx_lattice.ops.relations) GELU (class in mlx_lattice.nn.feature) gelu() (in module mlx_lattice.ops.feature) GeluApproxAttr (class in lattice_contract.dialect) generative_conv_transpose3d() (in module lattice_contract.dialect) (in module mlx_lattice.ops.conv) generative_kernel_relation() (in module mlx_lattice.ops.relations) generative_relation() (mlx_lattice.core.coords.manager.CoordinateManager method) GenerativeConvTranspose3d (class in mlx_lattice.nn.conv) global_avg_pool() (in module mlx_lattice.ops.pool) global_max_pool() (in module mlx_lattice.ops.pool) global_pool() (in module lattice_contract.dialect) (in module mlx_lattice.ops.pool) global_sum_pool() (in module mlx_lattice.ops.pool) GlobalAvgPool (class in mlx_lattice.nn.pool) GlobalMaxPool (class in mlx_lattice.nn.pool) GlobalSumPool (class in mlx_lattice.nn.pool) graph (mlx_lattice.artifact.io.LatticeArtifact attribute) (mlx_lattice.artifact.LatticeArtifact attribute) group_size (lattice_contract.mlir.builder.Packing attribute) (mlx_lattice.core.quantized.QuantizedWeight attribute) I id (mlx_lattice.core.coords.manager.CoordinateMapKey attribute) implicit_gemm (mlx_lattice.core.relations.views.KernelRelation attribute) in_channels (mlx_lattice.core.quantized.QuantizedWeight attribute) in_edge_ids (mlx_lattice.core.relations.views.KernelRelation property) in_row_offsets (mlx_lattice.core.relations.views.KernelRelation property) in_rows (mlx_lattice.core.relations.views.RelationEdges attribute) in_view (mlx_lattice.core.relations.views.KernelRelation property) input_csr (mlx_lattice.core.relations.views.KernelRelation attribute) insert_coords() (mlx_lattice.core.coords.manager.CoordinateManager method) interpolate_point_features() (in module mlx_lattice.core.coords.quantization) (in module mlx_lattice.ops.quantization) intersection_coords() (in module mlx_lattice.core.coords.set_ops) inverse_map() (in module mlx_lattice.core.coords.set_ops) (mlx_lattice.core.coords.manager.CoordinateManager method) inverse_rows (mlx_lattice.core.coords.ordering.CoordinateOrdering attribute) (mlx_lattice.core.coords.quantization.SparseQuantization attribute) ir_version (mlx_lattice.artifact.plan.RuntimePlan attribute) (mlx_lattice.artifact.RuntimePlan attribute) is_centered_submanifold (mlx_lattice.core.relations.specs.KernelSpec property) is_pointwise (mlx_lattice.core.quantized.QuantizedWeight property) (mlx_lattice.core.relations.specs.KernelSpec property) iter_ops() (lattice_contract.schema.DialectSchema method) J JoinAttr (class in lattice_contract.dialect) K kernel_csr (mlx_lattice.core.relations.views.KernelRelation attribute) kernel_edge_ids (mlx_lattice.core.relations.views.KernelRelation property) kernel_ids (mlx_lattice.core.relations.views.RelationEdges attribute) kernel_offsets (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.SparseRelationContract attribute) kernel_offsets() (in module mlx_lattice.core.coords.builders) (in module mlx_lattice.ops.relations) kernel_relation() (in module mlx_lattice.ops.relations) (mlx_lattice.core.coords.manager.CoordinateManager method) kernel_row_offsets (mlx_lattice.core.relations.views.KernelRelation property) kernel_size (mlx_lattice.core.quantized.QuantizedWeight attribute) kernel_view (mlx_lattice.core.relations.views.KernelRelation property) KernelRelation (class in mlx_lattice.core.relations.views) KernelSpec (class in mlx_lattice.core.relations.specs) kind (lattice_contract.mlir.builder.Packing attribute) (lattice_contract.schema.AttrParameter attribute) (lattice_contract.schema.OpAttributeDef attribute) (lattice_contract.schema.OperandDef attribute) (lattice_contract.schema.TypeParameter attribute) (mlx_lattice.core.relations.views.SparseRelationContract attribute) knn_relation() (in module mlx_lattice.ops.relations) L lattice_artifact_status() (in module mlx_lattice.artifact) (in module mlx_lattice.artifact.validation) lattice_contract.artifact module lattice_contract.dialect module lattice_contract.mlir.builder module lattice_contract.schema module lattice_graph_operation_names() (in module mlx_lattice.artifact) (in module mlx_lattice.artifact.validation) lattice_graph_status() (in module mlx_lattice.artifact) (in module mlx_lattice.artifact.validation) LatticeArtifact (class in mlx_lattice.artifact) (class in mlx_lattice.artifact.io) LatticeMLIRStatus (class in mlx_lattice.artifact) (class in mlx_lattice.artifact.validation) LatticeProgram (class in mlx_lattice.artifact) (class in mlx_lattice.artifact.runtime) layer_norm() (in module lattice_contract.dialect) (in module mlx_lattice.ops.feature) layer_norm_features() (in module mlx_lattice.ops.feature) LayerNorm (class in mlx_lattice.nn.feature) layout (mlx_lattice.core.quantized.QuantizedWeight attribute) leaky_relu() (in module mlx_lattice.ops.feature) LeakyReLU (class in mlx_lattice.nn.feature) lhs_rows (mlx_lattice.core.coords.alignment.SparseAlignment attribute) Linear (class in mlx_lattice.nn.feature) linear() (in module lattice_contract.dialect) (in module mlx_lattice.ops.feature) linear_features() (in module mlx_lattice.ops.feature) load_lattice_artifact() (in module mlx_lattice.artifact) (in module mlx_lattice.artifact.io) load_lattice_program() (in module mlx_lattice.artifact) (in module mlx_lattice.artifact.runtime) lookup_coords() (in module mlx_lattice.core.coords.set_ops) M manager_id (mlx_lattice.core.coords.manager.CoordinateMapKey attribute) mapped_conv_weight() (in module mlx_lattice.ops._relation_exec) max_neighbors (mlx_lattice.core.relations.views.NeighborRelation attribute) max_pool3d() (in module mlx_lattice.ops.pool) MaxPool3d (class in mlx_lattice.nn.pool) mlir() (lattice_contract.mlir.builder.Packing method) (lattice_contract.mlir.builder.SparseTensorType method) (lattice_contract.mlir.builder.TensorType method) (lattice_contract.mlir.builder.WeightType method) mlir_emitter() (in module lattice_contract.mlir.builder) MLIRModuleBuilder (class in lattice_contract.mlir.builder) mlx_lattice.artifact module mlx_lattice.artifact.io module mlx_lattice.artifact.plan module mlx_lattice.artifact.runtime module mlx_lattice.artifact.validation module mlx_lattice.core.coords.alignment module mlx_lattice.core.coords.builders module mlx_lattice.core.coords.manager module mlx_lattice.core.coords.occupancy module mlx_lattice.core.coords.ordering module mlx_lattice.core.coords.quantization module mlx_lattice.core.coords.set_ops module mlx_lattice.core.quantized module mlx_lattice.core.relations.specs module mlx_lattice.core.relations.views module mlx_lattice.core.tensor module mlx_lattice.core.types module mlx_lattice.nn.conv module mlx_lattice.nn.feature module mlx_lattice.nn.pool module mlx_lattice.nn.quantized_conv module mlx_lattice.nn.quantized_feature module mlx_lattice.ops._quantized module mlx_lattice.ops._relation_exec module mlx_lattice.ops.conv module mlx_lattice.ops.entropy module mlx_lattice.ops.feature module mlx_lattice.ops.pool module mlx_lattice.ops.quantization module mlx_lattice.ops.relations module mlx_lattice.ops.tensor module mnemonic (lattice_contract.schema.AttrDef attribute) (lattice_contract.schema.TypeDef attribute) mode (lattice_contract.mlir.builder.Packing attribute) module lattice_contract.artifact lattice_contract.dialect lattice_contract.mlir.builder lattice_contract.schema mlx_lattice.artifact mlx_lattice.artifact.io mlx_lattice.artifact.plan mlx_lattice.artifact.runtime mlx_lattice.artifact.validation mlx_lattice.core.coords.alignment mlx_lattice.core.coords.builders mlx_lattice.core.coords.manager mlx_lattice.core.coords.occupancy mlx_lattice.core.coords.ordering mlx_lattice.core.coords.quantization mlx_lattice.core.coords.set_ops mlx_lattice.core.quantized mlx_lattice.core.relations.specs mlx_lattice.core.relations.views mlx_lattice.core.tensor mlx_lattice.core.types mlx_lattice.nn.conv mlx_lattice.nn.feature mlx_lattice.nn.pool mlx_lattice.nn.quantized_conv mlx_lattice.nn.quantized_feature mlx_lattice.ops._quantized mlx_lattice.ops._relation_exec mlx_lattice.ops.conv mlx_lattice.ops.entropy mlx_lattice.ops.feature mlx_lattice.ops.pool mlx_lattice.ops.quantization mlx_lattice.ops.relations mlx_lattice.ops.tensor morton_codes() (in module mlx_lattice.core.coords.ordering) morton_order() (in module mlx_lattice.core.coords.ordering) morton_sort_coords() (in module mlx_lattice.core.coords.ordering) N n_in_capacity (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.SparseRelationContract attribute) n_kernels (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.SparseRelationContract attribute) n_out_capacity (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.SparseRelationContract attribute) n_query_capacity (mlx_lattice.core.relations.views.NeighborRelation attribute) n_source_capacity (mlx_lattice.core.relations.views.NeighborRelation attribute) name (lattice_contract.mlir.builder.SSAValue attribute) (lattice_contract.schema.AttrDef attribute) (lattice_contract.schema.AttrParameter attribute) (lattice_contract.schema.OpAttributeDef attribute) (lattice_contract.schema.OpDef attribute) (lattice_contract.schema.OperandDef attribute) (lattice_contract.schema.ResultDef attribute) (lattice_contract.schema.TypeDef attribute) (lattice_contract.schema.TypeParameter attribute) (mlx_lattice.artifact.plan.PlanArgument attribute) (mlx_lattice.artifact.plan.PlanOperation attribute) (mlx_lattice.artifact.plan.PlanOutput attribute) (mlx_lattice.artifact.plan.RuntimePlan attribute) (mlx_lattice.artifact.PlanArgument attribute) (mlx_lattice.artifact.PlanOperation attribute) (mlx_lattice.artifact.PlanOutput attribute) (mlx_lattice.artifact.RuntimePlan attribute) namespace (lattice_contract.schema.DialectSchema attribute) native_artifact_execution_available() (in module mlx_lattice.artifact) (in module mlx_lattice.artifact.runtime) NativeImplicitGemmView (in module mlx_lattice.core.coords.builders) NativeKernelRelation (in module mlx_lattice.core.coords.builders) NativeNeighborRelation (in module mlx_lattice.core.coords.builders) nbytes (mlx_lattice.core.quantized.QuantizedWeight property) neighbor_ids (mlx_lattice.core.relations.views.NeighborEdges attribute) NeighborEdges (class in mlx_lattice.core.relations.views) NeighborRelation (class in mlx_lattice.core.relations.views) normalized_cdf() (in module mlx_lattice.ops.entropy) normalized_conv_transpose3d() (in module lattice_contract.dialect) (in module mlx_lattice.ops.conv) normalized_generative_conv_transpose3d() (in module lattice_contract.dialect) (in module mlx_lattice.ops.conv) normalized_subm_conv3d() (in module lattice_contract.dialect) (in module mlx_lattice.ops.conv) NormalizedConvTranspose3d (class in mlx_lattice.nn.conv) NormalizedGenerativeConvTranspose3d (class in mlx_lattice.nn.conv) NormalizedSubmConv3d (class in mlx_lattice.nn.conv) O occupancy (mlx_lattice.core.coords.occupancy.SparseOccupancy attribute) occupancy_downsample() (in module mlx_lattice.core.coords.occupancy) occupancy_expand() (in module mlx_lattice.core.coords.occupancy) OccupancyExpansion (class in mlx_lattice.core.coords.occupancy) op() (lattice_contract.schema.DialectSchema method) op_attr() (in module lattice_contract.schema) op_by_python_name() (lattice_contract.schema.DialectSchema method) OpAttributeDef (class in lattice_contract.schema) OpDef (class in lattice_contract.schema) operand() (in module lattice_contract.schema) operand_types (mlx_lattice.artifact.plan.PlanOperation attribute) (mlx_lattice.artifact.PlanOperation attribute) OperandDef (class in lattice_contract.schema) operands (lattice_contract.schema.OpDef attribute) (mlx_lattice.artifact.plan.PlanOperation attribute) (mlx_lattice.artifact.PlanOperation attribute) OpHandle (in module lattice_contract.mlir.builder) ops (lattice_contract.schema.DialectSchema attribute) (mlx_lattice.artifact.plan.RuntimePlan attribute) (mlx_lattice.artifact.RuntimePlan attribute) optional (lattice_contract.schema.OperandDef attribute) order (mlx_lattice.core.coords.ordering.CoordinateOrdering attribute) out_channels (mlx_lattice.core.quantized.QuantizedWeight attribute) out_coords (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.SparseRelationContract attribute) out_count (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.SparseRelationContract property) out_in_map (mlx_lattice.core.relations.views.RelationImplicitGemmView attribute) out_rows (mlx_lattice.core.relations.views.RelationEdges attribute) out_view (mlx_lattice.core.relations.views.KernelRelation property) output_csr (mlx_lattice.core.relations.views.KernelRelation attribute) outputs (mlx_lattice.artifact.plan.RuntimePlan attribute) (mlx_lattice.artifact.RuntimePlan attribute) owns() (mlx_lattice.core.coords.manager.CoordinateManager method) P Packing (class in lattice_contract.mlir.builder) PackingAttr (class in lattice_contract.dialect) padding (mlx_lattice.core.relations.specs.KernelSpec attribute) (mlx_lattice.core.relations.views.SparseRelationContract attribute) parameters (lattice_contract.schema.AttrDef attribute) (lattice_contract.schema.TypeDef attribute) parent_rows (mlx_lattice.core.coords.occupancy.OccupancyExpansion attribute) plan (mlx_lattice.artifact.LatticeProgram attribute) (mlx_lattice.artifact.runtime.LatticeProgram attribute) PlanArgument (class in mlx_lattice.artifact) (class in mlx_lattice.artifact.plan) PlanOperation (class in mlx_lattice.artifact) (class in mlx_lattice.artifact.plan) PlanOutput (class in mlx_lattice.artifact) (class in mlx_lattice.artifact.plan) point_count (mlx_lattice.core.coords.quantization.PointVoxelMap property) PointInterpolationAttr (class in lattice_contract.dialect) PointVoxelInterpolation (in module mlx_lattice.core.coords.quantization) PointVoxelMap (class in mlx_lattice.core.coords.quantization) Pool3d (class in mlx_lattice.nn.pool) pool3d() (in module lattice_contract.dialect) (in module mlx_lattice.ops.pool) pool_transpose3d() (in module lattice_contract.dialect) (in module mlx_lattice.ops.pool) PoolModeAttr (class in lattice_contract.dialect) PoolTranspose3d (class in mlx_lattice.nn.pool) prune() (in module mlx_lattice.ops.tensor) prune_mask() (in module mlx_lattice.ops.tensor) python_name (lattice_contract.schema.OpDef attribute) Q qualified_op_name() (lattice_contract.schema.DialectSchema method) quantize_weight() (in module mlx_lattice.core.quantized) quantized_matmul() (in module mlx_lattice.ops._quantized) quantized_packing() (in module lattice_contract.mlir.builder) QuantizedConv3d (class in mlx_lattice.nn.quantized_conv) QuantizedConvTranspose3d (class in mlx_lattice.nn.quantized_conv) QuantizedGenerativeConvTranspose3d (class in mlx_lattice.nn.quantized_conv) QuantizedLinear (class in mlx_lattice.nn.quantized_feature) QuantizedSubmConv3d (class in mlx_lattice.nn.quantized_conv) QuantizedWeight (class in mlx_lattice.core.quantized) query_count (mlx_lattice.core.relations.views.NeighborRelation property) query_rows (mlx_lattice.core.relations.views.NeighborEdges attribute) R radius_relation() (in module mlx_lattice.ops.relations) range_decode() (in module mlx_lattice.ops.entropy) range_decode_from_prob() (in module mlx_lattice.ops.entropy) range_encode() (in module mlx_lattice.ops.entropy) range_encode_from_prob() (in module mlx_lattice.ops.entropy) rank (lattice_contract.mlir.builder.SparseTensorType attribute) rans_decode_from_prob() (in module mlx_lattice.ops.entropy) rans_encode_from_prob() (in module mlx_lattice.ops.entropy) ref() (lattice_contract.mlir.builder.SSAValue method) register() (lattice_contract.mlir.builder.EmitRegistry method) reindex_sparse() (in module mlx_lattice.ops.tensor) reindex_sparse_from_artifact() (in module mlx_lattice.ops.tensor) RelationCSRView (class in mlx_lattice.core.relations.views) RelationEdges (class in mlx_lattice.core.relations.views) RelationImplicitGemmView (class in mlx_lattice.core.relations.views) RelationSortedImplicitGemmView (class in mlx_lattice.core.relations.views) RelationView (in module mlx_lattice.core.relations.views) ReLU (class in mlx_lattice.nn.feature) relu() (in module mlx_lattice.ops.feature) reorder_rows (mlx_lattice.core.relations.views.RelationSortedImplicitGemmView attribute) replace() (mlx_lattice.core.tensor.SparseTensor method) replace_feature() (in module mlx_lattice.ops.tensor) require_implicit_gemm() (mlx_lattice.core.relations.views.KernelRelation method) require_sorted_implicit_gemm() (mlx_lattice.core.relations.views.KernelRelation method) required (lattice_contract.schema.OpAttributeDef attribute) resolve_op() (lattice_contract.schema.DialectSchema method) resolve_qualified_op() (lattice_contract.schema.DialectSchema method) result() (in module lattice_contract.schema) result_types (mlx_lattice.artifact.plan.PlanOperation attribute) (mlx_lattice.artifact.PlanOperation attribute) ResultDef (class in lattice_contract.schema) results (lattice_contract.schema.OpDef attribute) (mlx_lattice.artifact.plan.PlanOperation attribute) (mlx_lattice.artifact.PlanOperation attribute) return_() (lattice_contract.mlir.builder.MLIRModuleBuilder method) returns (mlx_lattice.artifact.plan.RuntimePlan attribute) (mlx_lattice.artifact.RuntimePlan attribute) reuse_coords_from() (mlx_lattice.core.tensor.SparseTensor method) rhs_rows (mlx_lattice.core.coords.alignment.SparseAlignment attribute) rms_norm() (in module lattice_contract.dialect) (in module mlx_lattice.ops.feature) rms_norm_features() (in module mlx_lattice.ops.feature) RMSNorm (class in mlx_lattice.nn.feature) role (mlx_lattice.artifact.plan.PlanArgument attribute) (mlx_lattice.artifact.plan.PlanOutput attribute) (mlx_lattice.artifact.PlanArgument attribute) (mlx_lattice.artifact.PlanOutput attribute) row_masks (mlx_lattice.core.relations.views.RelationImplicitGemmView attribute) row_offsets (mlx_lattice.core.relations.views.KernelRelation property) (mlx_lattice.core.relations.views.NeighborRelation attribute) (mlx_lattice.core.relations.views.RelationCSRView attribute) rows (mlx_lattice.core.coords.quantization.PointVoxelMap attribute) RuntimePlan (class in mlx_lattice.artifact) (class in mlx_lattice.artifact.plan) S same_coords() (mlx_lattice.core.tensor.SparseTensor method) save_lattice_artifact() (in module mlx_lattice.artifact) (in module mlx_lattice.artifact.io) scale_dtype (lattice_contract.mlir.builder.Packing attribute) scales (mlx_lattice.core.quantized.QuantizedWeight attribute) schema_digest (mlx_lattice.artifact.plan.RuntimePlan attribute) (mlx_lattice.artifact.RuntimePlan attribute) schema_digest() (in module lattice_contract.schema) schema_fingerprint() (in module lattice_contract.schema) shape (mlx_lattice.core.tensor.SparseTensor property) Sigmoid (class in mlx_lattice.nn.feature) sigmoid() (in module mlx_lattice.ops.feature) SiLU (class in mlx_lattice.nn.feature) silu() (in module mlx_lattice.ops.feature) size (mlx_lattice.core.relations.specs.KernelSpec attribute) softmax() (in module lattice_contract.dialect) Softplus (class in mlx_lattice.nn.feature) softplus() (in module mlx_lattice.ops.feature) sorted_implicit_gemm (mlx_lattice.core.relations.views.KernelRelation attribute) sorted_kv_out_in_map (mlx_lattice.core.relations.views.RelationSortedImplicitGemmView attribute) sorted_out_in_map (mlx_lattice.core.relations.views.RelationSortedImplicitGemmView attribute) source_active_rows (mlx_lattice.core.relations.views.SparseRelationContract attribute) source_coords (mlx_lattice.core.relations.views.SparseRelationContract attribute) source_rows (mlx_lattice.core.relations.views.NeighborEdges attribute) sparse_add() (in module mlx_lattice.ops.tensor) sparse_binary() (in module lattice_contract.dialect) (in module mlx_lattice.ops.tensor) sparse_binary_from_artifact() (in module mlx_lattice.ops.tensor) sparse_binary_op() (in module mlx_lattice.ops.tensor) sparse_cat() (in module lattice_contract.dialect) sparse_cat_aligned() (in module mlx_lattice.ops.tensor) sparse_cat_from_artifact() (in module mlx_lattice.ops.tensor) sparse_collate() (in module mlx_lattice.ops.tensor) sparse_conv_features_from_implicit_gemm_view() (in module mlx_lattice.ops._relation_exec) sparse_conv_features_from_relation() (in module mlx_lattice.ops._relation_exec) sparse_conv_features_sorted_direct_reference_from_relation() (in module mlx_lattice.ops._relation_exec) sparse_conv_features_sorted_from_relation() (in module mlx_lattice.ops._relation_exec) sparse_conv_normalization_denominator_from_relation() (in module mlx_lattice.ops._relation_exec) sparse_decompose() (in module lattice_contract.dialect) sparse_decompose_emitter() (in module lattice_contract.mlir.builder) sparse_decompose_from_artifact() (in module mlx_lattice.artifact.runtime) sparse_from_coordinates() (in module mlx_lattice.ops.tensor) sparse_make() (in module lattice_contract.dialect) sparse_make_from_artifact() (in module mlx_lattice.artifact.runtime) sparse_maximum() (in module mlx_lattice.ops.tensor) sparse_minimum() (in module mlx_lattice.ops.tensor) sparse_mul() (in module mlx_lattice.ops.tensor) sparse_pool_features_from_relation() (in module mlx_lattice.ops._relation_exec) sparse_quantize() (in module mlx_lattice.core.coords.quantization) sparse_quantized_conv_features_from_relation() (in module mlx_lattice.ops._relation_exec) sparse_reindex() (in module lattice_contract.dialect) sparse_sub() (in module mlx_lattice.ops.tensor) sparse_with_features() (in module lattice_contract.dialect) sparse_with_features_from_artifact() (in module mlx_lattice.ops.tensor) SparseAlignment (class in mlx_lattice.core.coords.alignment) SparseBinaryMode (in module mlx_lattice.ops.tensor) SparseJoin (in module mlx_lattice.core.coords.alignment) (in module mlx_lattice.ops.tensor) SparseOccupancy (class in mlx_lattice.core.coords.occupancy) SparseQuantization (class in mlx_lattice.core.coords.quantization) SparseRelationContract (class in mlx_lattice.core.relations.views) SparseTensor (class in mlx_lattice.core.tensor) SparseTensorComponents (class in mlx_lattice.core.tensor) SparseTensorType (class in lattice_contract.dialect) (class in lattice_contract.mlir.builder) SSAValue (class in lattice_contract.mlir.builder) storage_in_channels (mlx_lattice.core.quantized.QuantizedWeight property) stride (mlx_lattice.core.coords.manager.CoordinateMapKey attribute) (mlx_lattice.core.relations.specs.KernelSpec attribute) (mlx_lattice.core.relations.views.SparseRelationContract attribute) (mlx_lattice.core.tensor.SparseTensor attribute) (mlx_lattice.core.tensor.SparseTensorComponents attribute) subm_conv3d() (in module lattice_contract.dialect) (in module mlx_lattice.ops.conv) submanifold_kernel_relation() (in module mlx_lattice.ops.relations) (mlx_lattice.core.coords.manager.CoordinateManager method) SubmConv3d (class in mlx_lattice.nn.conv) sum_pool3d() (in module mlx_lattice.ops.pool) summary (lattice_contract.schema.AttrDef attribute) (lattice_contract.schema.OpDef attribute) (lattice_contract.schema.TypeDef attribute) SumPool3d (class in mlx_lattice.nn.pool) T Tanh (class in mlx_lattice.nn.feature) tanh() (in module mlx_lattice.ops.feature) target_active_rows (mlx_lattice.core.relations.views.SparseRelationContract attribute) target_conv3d() (in module lattice_contract.dialect) target_conv_transpose3d() (in module lattice_contract.dialect) target_coords (mlx_lattice.core.relations.views.SparseRelationContract attribute) target_kernel_relation() (in module mlx_lattice.ops.relations) (mlx_lattice.core.coords.manager.CoordinateManager method) target_normalized_conv_transpose3d() (in module lattice_contract.dialect) target_transposed_view() (mlx_lattice.core.coords.manager.CoordinateManager method) TensorType (class in lattice_contract.mlir.builder) text (lattice_contract.mlir.builder.TensorType attribute) tile_masks (mlx_lattice.core.relations.views.RelationSortedImplicitGemmView attribute) to_mlir() (lattice_contract.mlir.builder.MLIRModuleBuilder method) to_quantized() (mlx_lattice.nn.conv.Conv3d method) (mlx_lattice.nn.conv.ConvTranspose3d method) (mlx_lattice.nn.conv.GenerativeConvTranspose3d method) (mlx_lattice.nn.conv.SubmConv3d method) (mlx_lattice.nn.feature.Linear method) topk_rows() (in module mlx_lattice.ops.tensor) transposed_kernel_relation() (in module mlx_lattice.ops.relations) (mlx_lattice.core.coords.manager.CoordinateManager method) trilinear_upsample3d() (in module lattice_contract.dialect) (in module mlx_lattice.ops.pool) TrilinearUpsample3d (class in mlx_lattice.nn.pool) triple() (in module mlx_lattice.core.types) type (lattice_contract.mlir.builder.SSAValue attribute) (lattice_contract.schema.OperandDef attribute) (lattice_contract.schema.ResultDef attribute) (mlx_lattice.artifact.plan.PlanArgument attribute) (mlx_lattice.artifact.plan.PlanOutput attribute) (mlx_lattice.artifact.PlanArgument attribute) (mlx_lattice.artifact.PlanOutput attribute) type() (lattice_contract.schema.DialectSchema method) type_param() (in module lattice_contract.schema) TypeDef (class in lattice_contract.schema) TypeParameter (class in lattice_contract.schema) types (lattice_contract.schema.DialectSchema attribute) U union_coords() (in module mlx_lattice.core.coords.set_ops) V valid (mlx_lattice.artifact.LatticeMLIRStatus attribute) (mlx_lattice.artifact.validation.LatticeMLIRStatus attribute) validate_lattice_artifact() (in module mlx_lattice.artifact) (in module mlx_lattice.artifact.validation) validate_lattice_graph() (in module mlx_lattice.artifact) (in module mlx_lattice.artifact.validation) values (lattice_contract.schema.AttrDef attribute) volume (mlx_lattice.core.relations.specs.KernelSpec property) voxelize() (in module lattice_contract.dialect) (in module mlx_lattice.ops.quantization) voxelize_features() (in module mlx_lattice.core.coords.quantization) voxelize_with_quantization() (in module mlx_lattice.ops.quantization) VoxelReduction (in module mlx_lattice.core.coords.quantization) VoxelReductionAttr (class in lattice_contract.dialect) W weight (mlx_lattice.core.quantized.QuantizedWeight attribute) weight() (in module lattice_contract.dialect) weight_emitter() (in module lattice_contract.mlir.builder) weight_file (mlx_lattice.artifact.plan.RuntimePlan attribute) (mlx_lattice.artifact.RuntimePlan attribute) weight_from_artifact() (in module mlx_lattice.artifact.runtime) WeightLayoutAttr (class in lattice_contract.dialect) weights (mlx_lattice.artifact.io.LatticeArtifact attribute) (mlx_lattice.artifact.LatticeArtifact attribute) (mlx_lattice.artifact.LatticeProgram attribute) (mlx_lattice.artifact.runtime.LatticeProgram attribute) (mlx_lattice.core.coords.quantization.PointVoxelMap attribute) WeightType (class in lattice_contract.dialect) (class in lattice_contract.mlir.builder)