Index

A

activation() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) arity (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.pooling) AvgPool3d (class in torch_lattice.nn.modules.pooling)

B

backward (torch_lattice_bench.harness.BenchmarkCase attribute) batch_norm() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) batch_size (torch_lattice.artifact.io.LatticeModelArtifactOptions attribute) (torch_lattice_conformance.generate.SparseInputSpec attribute) BatchNorm (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.builder.PackedWeight attribute) build_kernel_map() (in module torch_lattice.nn.functional.conv.kmap.build_kmap) build_pool_output_coords() (in module torch_lattice.nn.functional.relation) build_target_out_in_map() (in module torch_lattice.nn.functional.relation)

C

C (torch_lattice.tensor.SparseTensor property) calc_ti_weights() (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) case (torch_lattice_bench.harness.BenchmarkResult attribute) cat() (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)

CodedCSR (torch_lattice.nn.functional.conv.conv_config.Dataflow attribute) CompatCase (class in torch_lattice_conformance.migration) Conv3d (class in torch_lattice.nn.modules.conv) conv3d() (in module torch_lattice.nn.functional.conv.conv) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) conv_transpose3d() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) convert_transposed_out_in_map() (in module torch_lattice.nn.functional.conv.hash.query) ConvMode (class in torch_lattice.nn.functional.conv.conv_mode) ConvTranspose3d (class in torch_lattice.nn.modules.conv) cpu() (torch_lattice.tensor.SparseTensor method) cuda() (torch_lattice.tensor.SparseTensor method) current (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder property)

D

Dataflow (class in torch_lattice.nn.functional.conv.conv_config) dense() (torch_lattice.tensor.SparseTensor method) dense_argument() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) dequantize_artifact_weight() (in module torch_lattice.artifact.builder) detach() (torch_lattice.tensor.SparseTensor method)

device (torch_lattice_bench.harness.BenchmarkResult attribute) devoxelize() (in module torch_lattice.nn.functional.devoxelize) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) dropout() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) dtype (torch_lattice_conformance.generate.SparseInputSpec attribute)

E

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)

F

F (torch_lattice.tensor.SparseTensor property) 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) flatten() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) fn (torch_lattice.artifact.builder.ModuleLowering attribute) forward() (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.conv.TargetConv3d 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.Pool3d method) (torch_lattice_conformance.e2e.GenerativeTransposeConvolution method) (torch_lattice_conformance.e2e.PointVoxel method) (torch_lattice_conformance.e2e.QuantizedClassifier method) (torch_lattice_conformance.e2e.SparseClassifier method) (torch_lattice_conformance.e2e.TargetBranch method) (torch_lattice_conformance.e2e.TransposeConvolution 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)

FuzzCase (class in torch_lattice_conformance.generate)

G

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

get_kmap_mode() (in module torch_lattice.nn.functional.conv.conv_mode) global_avg_pool() (in module torch_lattice.nn.functional.pooling) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) global_max_pool() (in module torch_lattice.nn.functional.pooling) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) global_pool() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) GlobalAvgPool (class in torch_lattice.nn.modules.pooling) GlobalMaxPool (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.modules.norm)

H

half() (torch_lattice.tensor.SparseTensor method)

I

identity() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) ImplicitGEMM (torch_lattice.nn.functional.conv.conv_config.Dataflow attribute) init() (in module torch_lattice.backends) input_dtype (torch_lattice.artifact.io.LatticeModelArtifactOptions attribute)

input_mode (torch_lattice_conformance.generate.FuzzCase attribute) input_stride (torch_lattice.artifact.io.LatticeModelArtifactOptions attribute) instance_norm() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) InstanceNorm (class in torch_lattice.nn.modules.norm) is_leaf_module() (torch_lattice.artifact.fx.LatticeTracer method)

K

keys_check() (in module torch_lattice.nn.functional.conv.conv_config)

kind (torch_lattice.artifact.builder.ArtifactValue 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.modules.norm)

leaky_relu() (in module torch_lattice.nn.functional.activation) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) LeakyReLU (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) matches() (torch_lattice.artifact.builder.ModuleLowering method) max_pool3d() (in module torch_lattice.nn.functional.pooling) MaxPool3d (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) Mode (in module torch_lattice_bench.harness) mode (torch_lattice_bench.harness.BenchmarkResult attribute) mode0 (torch_lattice.nn.functional.conv.conv_mode.ConvMode attribute) mode1 (torch_lattice.nn.functional.conv.conv_mode.ConvMode attribute) mode2 (torch_lattice.nn.functional.conv.conv_mode.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.backends 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.hash.hash torch_lattice.nn.functional.conv.hash.query 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)

notes (torch_lattice_bench.harness.BenchmarkResult attribute)

O

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) PackedWeight (class in torch_lattice.artifact.builder) 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.modules.pooling) pool3d() (in module torch_lattice.nn.functional.pooling) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) prepare (torch_lattice_bench.harness.BenchmarkCase attribute)

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

ReLU (class in torch_lattice.nn.modules.activation) relu() (in module torch_lattice.nn.functional.activation) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) repeats (torch_lattice_bench.harness.BenchmarkResult attribute) reset_parameters() (torch_lattice.nn.modules.bev.ToBEVConvolution method) (torch_lattice.nn.modules.bev.ToDenseBEVConvolution method)

rms_norm() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) RMSNorm (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

s (torch_lattice.tensor.SparseTensor property) 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.builder.PackedWeight attribute) seed (torch_lattice_conformance.generate.FuzzCase attribute) (torch_lattice_conformance.migration.CompatCase attribute) set_conv_mode() (in module torch_lattice.nn.functional.conv.conv_mode) set_downsample_mode() (in module torch_lattice.nn.functional.conv.conv_mode) set_global_conv_config() (in module torch_lattice.nn.functional.conv.conv_config) set_kmap_mode() (in module torch_lattice.nn.functional.conv.conv_mode) setup (torch_lattice_bench.harness.BenchmarkCase attribute) Sigmoid (class in torch_lattice.nn.modules.activation) sigmoid() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) SiLU (class in torch_lattice.nn.modules.activation) silu() (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.modules.activation) softplus() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) sparse_add() (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.operators) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) sparse_cat() (in module torch_lattice.operators) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) sparse_crop() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method)

sparse_input() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) sparse_maximum() (in module torch_lattice.operators) sparse_minimum() (in module torch_lattice.operators) sparse_mul() (in module torch_lattice.operators) sparse_quantize() (in module torch_lattice.utils.quantize) sparse_sub() (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.modules.crop) SparseInputSpec (class in torch_lattice_conformance.generate) SparseTensor (class in torch_lattice.tensor) spatial_shape (torch_lattice_conformance.generate.SparseInputSpec attribute) spcount() (in module torch_lattice.nn.functional.count) spcrop() (in module torch_lattice.nn.functional.crop) spdevoxelize() (in module torch_lattice.nn.functional.devoxelize) spdownsample() (in module torch_lattice.nn.functional.conv.kmap.downsample) sphash() (in module torch_lattice.nn.functional.conv.hash.hash) (in module torch_lattice.nn.functional.hash) sphashquery() (in module torch_lattice.nn.functional.conv.hash.query) (in module torch_lattice.nn.functional.query) spupsample_generative() (in module torch_lattice.nn.functional.conv.kmap.upsample) spvoxelize() (in module torch_lattice.nn.functional.voxelize) stride (torch_lattice_conformance.generate.SparseInputSpec attribute) subm_conv3d() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) SubmConv3d (class in torch_lattice.nn.modules.conv) sum_pool3d() (in module torch_lattice.nn.functional.pooling) SumPool3d (class in torch_lattice.nn.modules.pooling) supports() (torch_lattice_bench.harness.BenchmarkCase method)

T

Tanh (class in torch_lattice.nn.modules.activation) tanh() (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) target_conv3d() (in module torch_lattice.nn.functional.conv.conv) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method) TargetBranch (class in torch_lattice_conformance.e2e) TargetBranchModel (class in torch_lattice_conformance.generate) TargetConv3d (class in torch_lattice.nn.modules.conv) to() (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.modules.bev) ToBEVHeightCompression (class in torch_lattice.nn.modules.bev) ToBEVReduction (class in torch_lattice.nn.modules.bev) ToDenseBEVConvolution (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.backends 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.hash.hash module torch_lattice.nn.functional.conv.hash.query 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.conv.kmap.build_kmap) TransposeConvolution (class in torch_lattice_conformance.e2e) tune() (in module torch_lattice.utils.tune) types (torch_lattice.artifact.builder.ModuleLowering attribute)

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.nn.functional.voxelize) (torch_lattice.artifact.builder.TorchLatticeArtifactBuilder method)

W

warmup (torch_lattice_bench.harness.BenchmarkResult attribute) weight (torch_lattice.artifact.builder.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) workload (torch_lattice_bench.harness.BenchmarkResult attribute)