Index

A

activation() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) arities (torch_lattice.artifact.builder.ModuleLowering attribute) artifact_dir (torch_lattice.artifact.io.LatticeArtifactSaveResult attribute) ArtifactValue (class in torch_lattice.artifact.builder)

atol (torch_lattice_conformance.generate.FuzzCase attribute) avg_pool3d() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.pooling) AvgPool3d (class in torch_lattice.nn) (class in torch_lattice.nn.modules.pooling)

B

backward (torch_lattice_bench.harness.BenchmarkCase attribute) batch_counts (torch_lattice.core.coords.CoordinateMap attribute) batch_indices (torch_lattice.SparseTensor property) (torch_lattice.tensor.SparseTensor property) batch_norm() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) batch_rows (torch_lattice.SparseTensor property) (torch_lattice.tensor.SparseTensor property) batch_size (torch_lattice.artifact.io.LatticeModelArtifactOptions attribute) (torch_lattice_conformance.generate.SparseInputSpec attribute) BatchNorm (class in torch_lattice.nn) (class in torch_lattice.nn.modules.norm) BenchmarkCase (class in torch_lattice_bench.harness) BenchmarkResult (class in torch_lattice_bench.harness)

biases (torch_lattice.artifact.weights.PackedWeight attribute) build_kernel_map() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.conv) (in module torch_lattice.nn.functional.conv.kmap) (in module torch_lattice.nn.functional.conv.kmap.build_kmap) build_pool_output_coords() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.relation) build_target_out_in_map() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.relation) build_target_transposed_out_in_map() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.relation) build_transposed_output_coords() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.relation)

C

cached_hashmaps (torch_lattice.core.coords.CoordinateManager property) cached_relations (torch_lattice.core.coords.CoordinateManager property) calc_ti_weights() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.devoxelize) call_function() (torch_lattice.artifact.fx.LatticeArtifactInterpreter method) call_method() (torch_lattice.artifact.fx.LatticeArtifactInterpreter method) call_module() (torch_lattice.artifact.fx.LatticeArtifactInterpreter method) CanonicalKernelLayout (class in torch_lattice_conformance.e2e) case (torch_lattice_bench.harness.BenchmarkResult attribute) cat() (in module torch_lattice) (in module torch_lattice.operators) channels (torch_lattice.artifact.builder.ArtifactValue attribute) (torch_lattice_conformance.generate.SparseInputSpec attribute) clean (torch_lattice.artifact.io.LatticeModelArtifactOptions attribute) clear_global_conv_config() (in module torch_lattice.nn.functional.conv.conv_config) clear_relations() (torch_lattice.core.coords.CoordinateManager method) CompatCase (class in torch_lattice_conformance.migration) Conv3d (class in torch_lattice.nn) (class in torch_lattice.nn.modules.conv) conv3d() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.conv) (in module torch_lattice.nn.functional.conv.conv) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method)

conv_transpose3d() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) ConvConfig (class in torch_lattice.nn.functional.conv.conv_config) convert_transposed_out_in_map() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.query) ConvMode (class in torch_lattice.nn.functional) (class in torch_lattice.nn.functional.conv) (class in torch_lattice.nn.functional.conv.conv_mode) ConvTranspose3d (class in torch_lattice.nn) (class in torch_lattice.nn.modules.conv) CoordinateManager (class in torch_lattice.core.coords) CoordinateMap (class in torch_lattice.core.coords) CoordinateMapKey (class in torch_lattice.core.coords) coords (torch_lattice.core.coords.CoordinateMap attribute) copy() (torch_lattice.nn.functional.conv.conv_config.ConvConfig method) cpu() (torch_lattice.SparseTensor method) (torch_lattice.tensor.SparseTensor method) cuda() (torch_lattice.SparseTensor method) (torch_lattice.tensor.SparseTensor method) current (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder property)

D

Dataflow (class in torch_lattice.nn.functional) (class in torch_lattice.nn.functional.conv) (class in torch_lattice.nn.functional.conv.conv_config) dataflow (torch_lattice.nn.functional.conv.conv_config.ConvConfig attribute) decomposed_coordinates (torch_lattice.SparseTensor property) (torch_lattice.tensor.SparseTensor property) decomposed_coordinates_and_features (torch_lattice.SparseTensor property) (torch_lattice.tensor.SparseTensor property) decomposed_features (torch_lattice.SparseTensor property) (torch_lattice.tensor.SparseTensor property) dense() (torch_lattice.SparseTensor method) (torch_lattice.tensor.SparseTensor method)

dense_argument() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) dequantize_artifact_weight() (in module torch_lattice.artifact.weights) detach() (torch_lattice.SparseTensor method) (torch_lattice.tensor.SparseTensor method) device (torch_lattice_bench.harness.BenchmarkResult attribute) devoxelize() (in module torch_lattice) (in module torch_lattice.nn.functional.devoxelize) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) dilation (torch_lattice.core.coords.RelationKey attribute) downsample_mode (torch_lattice.nn.functional.conv.conv_config.ConvConfig attribute) dropout() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) dtype (torch_lattice_conformance.generate.SparseInputSpec attribute)

E

epsilon (torch_lattice.nn.functional.conv.conv_config.ConvConfig attribute) execution (torch_lattice.core.coords.RelationKey attribute) extra_repr() (torch_lattice.nn.modules.bev.ToBEVConvolution method) (torch_lattice.nn.modules.bev.ToBEVHeightCompression method) (torch_lattice.nn.modules.bev.ToBEVReduction method) (torch_lattice.nn.modules.bev.ToDenseBEVConvolution method) (torch_lattice.nn.ToBEVConvolution method) (torch_lattice.nn.ToBEVHeightCompression method) (torch_lattice.nn.ToBEVReduction method) (torch_lattice.nn.ToDenseBEVConvolution method)

F

family (torch_lattice_conformance.generate.FuzzCase attribute) (torch_lattice_conformance.migration.CompatCase attribute) fapply() (in module torch_lattice.nn.utils.apply) FetchOnDemand (torch_lattice.nn.functional.conv.conv_config.Dataflow attribute) (torch_lattice.nn.functional.conv.Dataflow attribute) (torch_lattice.nn.functional.Dataflow attribute) flatten() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) fn (torch_lattice.artifact.builder.ModuleLowering attribute) FOD_fusion (torch_lattice.nn.functional.conv.conv_config.ConvConfig attribute) forward() (torch_lattice.nn.BatchNorm method) (torch_lattice.nn.GELU method) (torch_lattice.nn.GlobalAvgPool method) (torch_lattice.nn.GlobalMaxPool method) (torch_lattice.nn.GlobalSumPool method) (torch_lattice.nn.GroupNorm method) (torch_lattice.nn.InstanceNorm method) (torch_lattice.nn.LayerNorm method) (torch_lattice.nn.LeakyReLU method) (torch_lattice.nn.modules.activation.GELU method) (torch_lattice.nn.modules.activation.LeakyReLU method) (torch_lattice.nn.modules.activation.ReLU method) (torch_lattice.nn.modules.activation.Sigmoid method) (torch_lattice.nn.modules.activation.SiLU method) (torch_lattice.nn.modules.activation.Softplus method) (torch_lattice.nn.modules.activation.Tanh method) (torch_lattice.nn.modules.bev.ToBEVConvolution method) (torch_lattice.nn.modules.bev.ToBEVHeightCompression method) (torch_lattice.nn.modules.bev.ToBEVReduction method) (torch_lattice.nn.modules.bev.ToDenseBEVConvolution method) (torch_lattice.nn.modules.crop.SparseCrop method) (torch_lattice.nn.modules.norm.BatchNorm method) (torch_lattice.nn.modules.norm.GroupNorm method) (torch_lattice.nn.modules.norm.InstanceNorm method) (torch_lattice.nn.modules.norm.LayerNorm method) (torch_lattice.nn.modules.norm.RMSNorm method) (torch_lattice.nn.modules.pooling.GlobalAvgPool method) (torch_lattice.nn.modules.pooling.GlobalMaxPool method) (torch_lattice.nn.modules.pooling.GlobalSumPool method) (torch_lattice.nn.modules.pooling.Pool3d method) (torch_lattice.nn.modules.pooling.PoolTranspose3d method) (torch_lattice.nn.modules.pooling.TrilinearUpsample3d method) (torch_lattice.nn.Pool3d method) (torch_lattice.nn.PoolTranspose3d method) (torch_lattice.nn.ReLU method) (torch_lattice.nn.RMSNorm method) (torch_lattice.nn.Sigmoid method) (torch_lattice.nn.SiLU method) (torch_lattice.nn.Softplus method) (torch_lattice.nn.SparseCrop method) (torch_lattice.nn.Tanh method) (torch_lattice.nn.ToBEVConvolution method) (torch_lattice.nn.ToBEVHeightCompression method) (torch_lattice.nn.ToBEVReduction method) (torch_lattice.nn.ToDenseBEVConvolution method) (torch_lattice.nn.TrilinearUpsample3d method) (torch_lattice_conformance.e2e.CanonicalKernelLayout method) (torch_lattice_conformance.e2e.GameleonReproductionBlock method) (torch_lattice_conformance.e2e.GenerativeTransposeConvolution method) (torch_lattice_conformance.e2e.NormalizedConvolution method) (torch_lattice_conformance.e2e.PointVoxel method) (torch_lattice_conformance.e2e.PoolTranspose method) (torch_lattice_conformance.e2e.QuantizedClassifier method) (torch_lattice_conformance.e2e.SparseClassifier method) (torch_lattice_conformance.e2e.SparseReindex method) (torch_lattice_conformance.e2e.TargetBranch method) (torch_lattice_conformance.e2e.TargetTransposeConvolution method) (torch_lattice_conformance.e2e.TransposeConvolution method) (torch_lattice_conformance.e2e.TrilinearUpsample method) (torch_lattice_conformance.generate.PointVoxelRoundTrip method) (torch_lattice_conformance.generate.SparseBranchModel method) (torch_lattice_conformance.generate.SparseChainModel method) (torch_lattice_conformance.generate.TargetBranchModel method)

forward_relation() (torch_lattice.core.coords.CoordinateManager method) FuzzCase (class in torch_lattice_conformance.generate)

G

GameleonReproductionBlock (class in torch_lattice_conformance.e2e) gather_scatter_kmap_from_out_in_map() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.relation) GatherScatter (torch_lattice.nn.functional.conv.conv_config.Dataflow attribute) (torch_lattice.nn.functional.conv.Dataflow attribute) (torch_lattice.nn.functional.Dataflow attribute) GELU (class in torch_lattice.nn) (class in torch_lattice.nn.modules.activation) gelu() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) generate() (in module torch_lattice_conformance.e2e) generated_target() (torch_lattice.core.coords.CoordinateManager method) generative_add() (in module torch_lattice) (in module torch_lattice.operators) generative_conv_transpose3d() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) GenerativeConvTranspose3d (class in torch_lattice.nn) (class in torch_lattice.nn.modules.conv) GenerativeTransposeConvolution (class in torch_lattice_conformance.e2e) get() (torch_lattice.core.coords.CoordinateManager method) get_conv_mode() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.conv) (in module torch_lattice.nn.functional.conv.conv_mode) get_default_conv_config() (in module torch_lattice.nn.functional.conv.conv_config) get_global_conv_config() (in module torch_lattice.nn.functional.conv.conv_config) get_kernel_offsets() (in module torch_lattice.nn.utils.kernel)

global_avg_pool() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.pooling) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) global_max_pool() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.pooling) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) global_pool() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.pooling) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) global_sum_pool() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.pooling) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) GlobalAvgPool (class in torch_lattice.nn) (class in torch_lattice.nn.modules.pooling) GlobalMaxPool (class in torch_lattice.nn) (class in torch_lattice.nn.modules.pooling) GlobalSumPool (class in torch_lattice.nn) (class in torch_lattice.nn.modules.pooling) graph_path (torch_lattice.artifact.io.LatticeArtifactSaveResult attribute) group (torch_lattice_bench.harness.BenchmarkCase attribute) (torch_lattice_bench.harness.BenchmarkResult attribute) group_norm() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) GroupNorm (class in torch_lattice.nn) (class in torch_lattice.nn.modules.norm)

H

half() (torch_lattice.SparseTensor method) (torch_lattice.tensor.SparseTensor method)

hashmap() (torch_lattice.core.coords.CoordinateManager method)

I

id (torch_lattice.core.coords.CoordinateMapKey attribute) identity() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) ifsort (torch_lattice.nn.functional.conv.conv_config.ConvConfig attribute) IGEMM_center_only (torch_lattice.nn.functional.conv.conv_config.ConvConfig attribute) ImplicitGEMM (torch_lattice.nn.functional.conv.conv_config.Dataflow attribute) (torch_lattice.nn.functional.conv.Dataflow attribute) (torch_lattice.nn.functional.Dataflow attribute) init() (in module torch_lattice.backends)

input_dtype (torch_lattice.artifact.io.LatticeModelArtifactOptions attribute) input_mode (torch_lattice_conformance.generate.FuzzCase attribute) insert() (torch_lattice.core.coords.CoordinateManager method) instance_norm() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) InstanceNorm (class in torch_lattice.nn) (class in torch_lattice.nn.modules.norm) inverse_relation() (torch_lattice.core.coords.CoordinateManager method) is_leaf_module() (torch_lattice.artifact.fx.LatticeTracer method)

K

kernel_size (torch_lattice.core.coords.RelationKey attribute)

kind (torch_lattice.artifact.builder.ArtifactValue attribute) kmap_mode (torch_lattice.nn.functional.conv.conv_config.ConvConfig attribute)

L

LatticeArtifactInterpreter (class in torch_lattice.artifact.fx) LatticeArtifactSaveResult (class in torch_lattice.artifact.io) LatticeModelArtifactError LatticeModelArtifactOptions (class in torch_lattice.artifact.io) LatticeTracer (class in torch_lattice.artifact.fx) layer_norm() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) LayerNorm (class in torch_lattice.nn) (class in torch_lattice.nn.modules.norm)

leaky_relu() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.activation) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) LeakyReLU (class in torch_lattice.nn) (class in torch_lattice.nn.modules.activation) linear() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) lower() (torch_lattice.artifact.builder.ModuleLowering method) lower_fx_artifact() (in module torch_lattice.artifact.fx) lower_module() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method)

M

main() (in module torch_lattice_bench.run) (in module torch_lattice_conformance.cli) (in module torch_lattice_conformance.e2e) (in module torch_lattice_conformance.generate) (in module torch_lattice_conformance.migration) manager_id (torch_lattice.core.coords.CoordinateMapKey attribute) max_pool3d() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.pooling) MaxPool3d (class in torch_lattice.nn) (class in torch_lattice.nn.modules.pooling) median_ms (torch_lattice_bench.harness.BenchmarkResult attribute) metadata (torch_lattice_conformance.generate.FuzzCase attribute) metrics (torch_lattice_bench.harness.BenchmarkCase attribute) min_ms (torch_lattice_bench.harness.BenchmarkResult attribute) mm_thresh (torch_lattice.nn.functional.conv.conv_config.ConvConfig attribute) Mode (in module torch_lattice_bench.harness) mode (torch_lattice_bench.harness.BenchmarkResult attribute) mode0 (torch_lattice.nn.functional.conv.conv_mode.ConvMode attribute) (torch_lattice.nn.functional.conv.ConvMode attribute) (torch_lattice.nn.functional.ConvMode attribute) mode1 (torch_lattice.nn.functional.conv.conv_mode.ConvMode attribute) (torch_lattice.nn.functional.conv.ConvMode attribute) (torch_lattice.nn.functional.ConvMode attribute) mode2 (torch_lattice.nn.functional.conv.conv_mode.ConvMode attribute) (torch_lattice.nn.functional.conv.ConvMode attribute) (torch_lattice.nn.functional.ConvMode attribute) modes (torch_lattice_bench.harness.BenchmarkCase attribute) module torch_lattice torch_lattice.artifact.builder torch_lattice.artifact.fx torch_lattice.artifact.io torch_lattice.artifact.weights torch_lattice.backends torch_lattice.core.coords torch_lattice.nn torch_lattice.nn.functional torch_lattice.nn.functional.activation torch_lattice.nn.functional.conv torch_lattice.nn.functional.conv.conv torch_lattice.nn.functional.conv.conv_config torch_lattice.nn.functional.conv.conv_mode torch_lattice.nn.functional.conv.kmap torch_lattice.nn.functional.conv.kmap.build_kmap torch_lattice.nn.functional.conv.kmap.downsample torch_lattice.nn.functional.conv.kmap.upsample torch_lattice.nn.functional.count torch_lattice.nn.functional.crop torch_lattice.nn.functional.devoxelize torch_lattice.nn.functional.hash torch_lattice.nn.functional.pooling torch_lattice.nn.functional.query torch_lattice.nn.functional.relation torch_lattice.nn.functional.voxelize torch_lattice.nn.modules.activation torch_lattice.nn.modules.bev torch_lattice.nn.modules.conv torch_lattice.nn.modules.crop torch_lattice.nn.modules.norm torch_lattice.nn.modules.pooling torch_lattice.nn.utils.apply torch_lattice.nn.utils.kernel torch_lattice.operators torch_lattice.tensor torch_lattice.utils.quantize torch_lattice.utils.to_dense torch_lattice.utils.tune torch_lattice_bench.catalog torch_lattice_bench.harness torch_lattice_bench.run torch_lattice_conformance.cli torch_lattice_conformance.e2e torch_lattice_conformance.generate torch_lattice_conformance.migration

module() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) module_lowering() (in module torch_lattice.artifact.builder) ModuleLowering (class in torch_lattice.artifact.builder)

N

name (torch_lattice_bench.harness.BenchmarkCase attribute) (torch_lattice_conformance.generate.FuzzCase attribute) normalized_conv3d() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.conv) (in module torch_lattice.nn.functional.conv.conv) normalized_conv_transpose3d() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) normalized_generative_conv_transpose3d() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) normalized_subm_conv3d() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method)

NormalizedConvolution (class in torch_lattice_conformance.e2e) NormalizedConvTranspose3d (class in torch_lattice.nn) (class in torch_lattice.nn.modules.conv) NormalizedGenerativeConvTranspose3d (class in torch_lattice.nn) (class in torch_lattice.nn.modules.conv) NormalizedSubmConv3d (class in torch_lattice.nn) (class in torch_lattice.nn.modules.conv) notes (torch_lattice_bench.harness.BenchmarkResult attribute)

O

operation (torch_lattice.core.coords.RelationKey attribute) output() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method)

output_kind (torch_lattice_conformance.generate.FuzzCase attribute) (torch_lattice_conformance.migration.CompatCase attribute)

P

p90_ms (torch_lattice_bench.harness.BenchmarkResult attribute) p95_ms (torch_lattice_bench.harness.BenchmarkResult attribute) pack_quantized_weight() (in module torch_lattice.artifact.weights) PackedWeight (class in torch_lattice.artifact.weights) padding (torch_lattice.core.coords.RelationKey attribute) params (torch_lattice_bench.harness.BenchmarkCase attribute) (torch_lattice_bench.harness.BenchmarkResult attribute) PointVoxel (class in torch_lattice_conformance.e2e) PointVoxelRoundTrip (class in torch_lattice_conformance.generate) Pool3d (class in torch_lattice.nn) (class in torch_lattice.nn.modules.pooling) pool3d() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.pooling) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method)

pool_transpose3d() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.pooling) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) PoolTranspose (class in torch_lattice_conformance.e2e) PoolTranspose3d (class in torch_lattice.nn) (class in torch_lattice.nn.modules.pooling) prepare (torch_lattice_bench.harness.BenchmarkCase attribute) prune() (in module torch_lattice) (in module torch_lattice.operators) prune_mask() (in module torch_lattice) (in module torch_lattice.operators)

Q

quantize_bits (torch_lattice.artifact.io.LatticeModelArtifactOptions attribute) quantize_group_size (torch_lattice.artifact.io.LatticeModelArtifactOptions attribute)

quantize_scale_dtype (torch_lattice.artifact.io.LatticeModelArtifactOptions attribute) QuantizedClassifier (class in torch_lattice_conformance.e2e)

R

reindex_sparse() (in module torch_lattice) (in module torch_lattice.operators) relation() (torch_lattice.core.coords.CoordinateManager method) relation_count (torch_lattice.core.coords.CoordinateManager property) RelationKey (class in torch_lattice.core.coords) ReLU (class in torch_lattice.nn) (class in torch_lattice.nn.modules.activation) relu() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.activation) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) repeats (torch_lattice_bench.harness.BenchmarkResult attribute) replace() (torch_lattice.SparseTensor method) (torch_lattice.tensor.SparseTensor method)

reset_parameters() (torch_lattice.nn.modules.bev.ToBEVConvolution method) (torch_lattice.nn.modules.bev.ToDenseBEVConvolution method) (torch_lattice.nn.ToBEVConvolution method) (torch_lattice.nn.ToDenseBEVConvolution method) rms_norm() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) RMSNorm (class in torch_lattice.nn) (class in torch_lattice.nn.modules.norm) rows_per_batch (torch_lattice_conformance.generate.SparseInputSpec attribute) rtol (torch_lattice_conformance.generate.FuzzCase attribute) run (torch_lattice_bench.harness.BenchmarkCase attribute) run_case() (in module torch_lattice_bench.harness) run_cases() (in module torch_lattice_bench.harness)

S

samples_ms (torch_lattice_bench.harness.BenchmarkResult attribute) save() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) save_lattice_model_artifact() (in module torch_lattice.artifact.io) scales (torch_lattice.artifact.weights.PackedWeight attribute) seed (torch_lattice_conformance.generate.FuzzCase attribute) (torch_lattice_conformance.migration.CompatCase attribute) set_conv_mode() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.conv) (in module torch_lattice.nn.functional.conv.conv_mode) set_forward_relation() (torch_lattice.core.coords.CoordinateManager method) set_generated_target() (torch_lattice.core.coords.CoordinateManager method) set_global_conv_config() (in module torch_lattice.nn.functional.conv.conv_config) set_hashmap() (torch_lattice.core.coords.CoordinateManager method) set_inverse_relation() (torch_lattice.core.coords.CoordinateManager method) set_relation() (torch_lattice.core.coords.CoordinateManager method) setup (torch_lattice_bench.harness.BenchmarkCase attribute) Sigmoid (class in torch_lattice.nn) (class in torch_lattice.nn.modules.activation) sigmoid() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) SiLU (class in torch_lattice.nn) (class in torch_lattice.nn.modules.activation) silu() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.activation) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) SkipCase skipped (torch_lattice_bench.harness.BenchmarkResult attribute) Softplus (class in torch_lattice.nn) (class in torch_lattice.nn.modules.activation) softplus() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) source (torch_lattice.core.coords.RelationKey attribute) sparse_add() (in module torch_lattice) (in module torch_lattice.operators) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) sparse_argument() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) sparse_binary() (in module torch_lattice) (in module torch_lattice.operators) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) sparse_cat() (in module torch_lattice) (in module torch_lattice.operators) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) sparse_crop() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) sparse_from_coordinates() (in module torch_lattice) (in module torch_lattice.operators) sparse_input() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) sparse_maximum() (in module torch_lattice) (in module torch_lattice.operators) sparse_minimum() (in module torch_lattice) (in module torch_lattice.operators) sparse_mul() (in module torch_lattice) (in module torch_lattice.operators)

sparse_quantize() (in module torch_lattice.utils.quantize) sparse_reindex() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) sparse_sub() (in module torch_lattice) (in module torch_lattice.operators) SparseBranchModel (class in torch_lattice_conformance.generate) SparseChainModel (class in torch_lattice_conformance.generate) SparseClassifier (class in torch_lattice_conformance.e2e) SparseCrop (class in torch_lattice.nn) (class in torch_lattice.nn.modules.crop) SparseInputSpec (class in torch_lattice_conformance.generate) SparseReindex (class in torch_lattice_conformance.e2e) SparseTensor (class in torch_lattice) (class in torch_lattice.tensor) spatial_range (torch_lattice.core.coords.CoordinateMap attribute) spatial_shape (torch_lattice_conformance.generate.SparseInputSpec attribute) spcount() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.count) spcrop() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.crop) spdevoxelize() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.devoxelize) spdownsample() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.conv) (in module torch_lattice.nn.functional.conv.kmap) (in module torch_lattice.nn.functional.conv.kmap.downsample) sphash() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.hash) sphashquery() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.query) split_mask_num (torch_lattice.nn.functional.conv.conv_config.ConvConfig attribute) split_mask_num_bwd (torch_lattice.nn.functional.conv.conv_config.ConvConfig attribute) spupsample_generative() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.conv) (in module torch_lattice.nn.functional.conv.kmap) (in module torch_lattice.nn.functional.conv.kmap.upsample) spvoxelize() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.voxelize) stride (torch_lattice.core.coords.CoordinateMapKey attribute) (torch_lattice.core.coords.RelationKey attribute) (torch_lattice_conformance.generate.SparseInputSpec attribute) subm_conv3d() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) SubmConv3d (class in torch_lattice.nn) (class in torch_lattice.nn.modules.conv) sum_pool3d() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.pooling) SumPool3d (class in torch_lattice.nn) (class in torch_lattice.nn.modules.pooling) supports() (torch_lattice_bench.harness.BenchmarkCase method)

T

Tanh (class in torch_lattice.nn) (class in torch_lattice.nn.modules.activation) tanh() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) target (torch_lattice.core.coords.RelationKey attribute) TargetBranch (class in torch_lattice_conformance.e2e) TargetBranchModel (class in torch_lattice_conformance.generate) TargetTransposeConvolution (class in torch_lattice_conformance.e2e) to() (torch_lattice.SparseTensor method) (torch_lattice.tensor.SparseTensor method) to_dense() (in module torch_lattice.utils.to_dense) to_json() (torch_lattice_bench.harness.BenchmarkResult method) to_mlir() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) ToBEVConvolution (class in torch_lattice.nn) (class in torch_lattice.nn.modules.bev) ToBEVHeightCompression (class in torch_lattice.nn) (class in torch_lattice.nn.modules.bev) ToBEVReduction (class in torch_lattice.nn) (class in torch_lattice.nn.modules.bev) ToDenseBEVConvolution (class in torch_lattice.nn) (class in torch_lattice.nn.modules.bev) torch_lattice module torch_lattice.artifact.builder module torch_lattice.artifact.fx module torch_lattice.artifact.io module torch_lattice.artifact.weights module torch_lattice.backends module torch_lattice.core.coords module torch_lattice.nn module torch_lattice.nn.functional module torch_lattice.nn.functional.activation module torch_lattice.nn.functional.conv module torch_lattice.nn.functional.conv.conv module torch_lattice.nn.functional.conv.conv_config module torch_lattice.nn.functional.conv.conv_mode module torch_lattice.nn.functional.conv.kmap module torch_lattice.nn.functional.conv.kmap.build_kmap module torch_lattice.nn.functional.conv.kmap.downsample module torch_lattice.nn.functional.conv.kmap.upsample module torch_lattice.nn.functional.count module torch_lattice.nn.functional.crop module torch_lattice.nn.functional.devoxelize module torch_lattice.nn.functional.hash module

torch_lattice.nn.functional.pooling module torch_lattice.nn.functional.query module torch_lattice.nn.functional.relation module torch_lattice.nn.functional.voxelize module torch_lattice.nn.modules.activation module torch_lattice.nn.modules.bev module torch_lattice.nn.modules.conv module torch_lattice.nn.modules.crop module torch_lattice.nn.modules.norm module torch_lattice.nn.modules.pooling module torch_lattice.nn.utils.apply module torch_lattice.nn.utils.kernel module torch_lattice.operators module torch_lattice.tensor module torch_lattice.utils.quantize module torch_lattice.utils.to_dense module torch_lattice.utils.tune module torch_lattice_bench.catalog module torch_lattice_bench.harness module torch_lattice_bench.run module torch_lattice_conformance.cli module torch_lattice_conformance.e2e module torch_lattice_conformance.generate module torch_lattice_conformance.migration module TorchLatticeArtifactBuilder (class in torch_lattice.artifact.builder) transpose_kernel_map() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.conv) (in module torch_lattice.nn.functional.conv.kmap) (in module torch_lattice.nn.functional.conv.kmap.build_kmap) TransposeConvolution (class in torch_lattice_conformance.e2e) trilinear_upsample3d() (in module torch_lattice.nn.functional) (in module torch_lattice.nn.functional.pooling) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) TrilinearUpsample (class in torch_lattice_conformance.e2e) TrilinearUpsample3d (class in torch_lattice.nn) (class in torch_lattice.nn.modules.pooling) tune() (in module torch_lattice) (in module torch_lattice.utils.tune)

U

units (torch_lattice_bench.harness.BenchmarkCase attribute) (torch_lattice_bench.harness.BenchmarkResult attribute)

V

validate (torch_lattice.artifact.io.LatticeModelArtifactOptions attribute) value (torch_lattice.artifact.builder.ArtifactValue attribute)

voxelize() (in module torch_lattice) (in module torch_lattice.nn.functional.voxelize) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method)

W

warmup (torch_lattice_bench.harness.BenchmarkResult attribute) weight (torch_lattice.artifact.weights.PackedWeight attribute) weight_keys (torch_lattice.artifact.io.LatticeArtifactSaveResult attribute) weights (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder property)

weights_path (torch_lattice.artifact.io.LatticeArtifactSaveResult attribute) wgrad_split_k (torch_lattice.nn.functional.conv.conv_config.ConvConfig attribute) with_coordinates() (torch_lattice.SparseTensor method) (torch_lattice.tensor.SparseTensor method) workload (torch_lattice_bench.harness.BenchmarkResult attribute)