mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-05-06 03:27:53 -05:00
ee9355584f
Fixes after merge conflicts: - Fix tensor reshape error message format - Fix __init__.py imports (remove BatchNorm2d, fix enable_autograd call) - Fix attention mask broadcasting for multi-head attention - Fix memoization module to use matmul instead of @ operator - Fix capstone module count_parameters and CosineSchedule usage - Add missing imports to benchmark.py (dataclass, Profiler, platform, os) - Simplify capstone pipeline test to avoid data shape mismatch All 20 modules now pass tito test --all
986 lines
133 KiB
Python
Generated
986 lines
133 KiB
Python
Generated
# ╔═══════════════════════════════════════════════════════════════════════════════╗
|
|
# ║ 🚨 CRITICAL WARNING 🚨 ║
|
|
# ║ AUTOGENERATED! DO NOT EDIT! ║
|
|
# ║ ║
|
|
# ║ This file is AUTOMATICALLY GENERATED from source modules. ║
|
|
# ║ ANY CHANGES MADE HERE WILL BE LOST when modules are re-exported! ║
|
|
# ║ ║
|
|
# ║ ✅ TO EDIT: src/[unknown]/[unknown].py ║
|
|
# ║ ✅ TO EXPORT: Run 'tito module complete <module_name>' ║
|
|
# ║ ║
|
|
# ║ 🛡️ STUDENT PROTECTION: This file contains optimized implementations. ║
|
|
# ║ Editing it directly may break module functionality and training. ║
|
|
# ║ ║
|
|
# ║ 🎓 LEARNING TIP: Work in src/ (developers) or modules/ (learners) ║
|
|
# ║ The tinytorch/ directory is generated code - edit source files instead! ║
|
|
# ╚═══════════════════════════════════════════════════════════════════════════════╝
|
|
# Autogenerated by nbdev
|
|
|
|
d = { 'settings': { 'branch': 'main',
|
|
'doc_baseurl': '/TinyTorch/',
|
|
'doc_host': 'https://tinytorch.github.io',
|
|
'git_url': 'https://github.com/tinytorch/TinyTorch/',
|
|
'lib_path': 'tinytorch'},
|
|
'syms': { 'tinytorch.applications.tinygpt': { 'tinytorch.applications.tinygpt.CompleteTinyGPTPipeline': ( '20_capstone/capstone.html#completetinygptpipeline',
|
|
'tinytorch/applications/tinygpt.py'),
|
|
'tinytorch.applications.tinygpt.CompleteTinyGPTPipeline.__init__': ( '20_capstone/capstone.html#completetinygptpipeline.__init__',
|
|
'tinytorch/applications/tinygpt.py'),
|
|
'tinytorch.applications.tinygpt.CompleteTinyGPTPipeline.generate_text': ( '20_capstone/capstone.html#completetinygptpipeline.generate_text',
|
|
'tinytorch/applications/tinygpt.py'),
|
|
'tinytorch.applications.tinygpt.CompleteTinyGPTPipeline.optimize_model': ( '20_capstone/capstone.html#completetinygptpipeline.optimize_model',
|
|
'tinytorch/applications/tinygpt.py'),
|
|
'tinytorch.applications.tinygpt.CompleteTinyGPTPipeline.prepare_training_data': ( '20_capstone/capstone.html#completetinygptpipeline.prepare_training_data',
|
|
'tinytorch/applications/tinygpt.py'),
|
|
'tinytorch.applications.tinygpt.CompleteTinyGPTPipeline.train': ( '20_capstone/capstone.html#completetinygptpipeline.train',
|
|
'tinytorch/applications/tinygpt.py'),
|
|
'tinytorch.applications.tinygpt.TinyGPT': ( '20_capstone/capstone.html#tinygpt',
|
|
'tinytorch/applications/tinygpt.py'),
|
|
'tinytorch.applications.tinygpt.TinyGPT.__init__': ( '20_capstone/capstone.html#tinygpt.__init__',
|
|
'tinytorch/applications/tinygpt.py'),
|
|
'tinytorch.applications.tinygpt.TinyGPTTrainer': ( '20_capstone/capstone.html#tinygpttrainer',
|
|
'tinytorch/applications/tinygpt.py'),
|
|
'tinytorch.applications.tinygpt.TinyGPTTrainer.__init__': ( '20_capstone/capstone.html#tinygpttrainer.__init__',
|
|
'tinytorch/applications/tinygpt.py'),
|
|
'tinytorch.applications.tinygpt.TinyGPTTrainer.prepare_batch': ( '20_capstone/capstone.html#tinygpttrainer.prepare_batch',
|
|
'tinytorch/applications/tinygpt.py'),
|
|
'tinytorch.applications.tinygpt.TinyGPTTrainer.train_step': ( '20_capstone/capstone.html#tinygpttrainer.train_step',
|
|
'tinytorch/applications/tinygpt.py'),
|
|
'tinytorch.applications.tinygpt.test_unit_complete_pipeline': ( '20_capstone/capstone.html#test_unit_complete_pipeline',
|
|
'tinytorch/applications/tinygpt.py'),
|
|
'tinytorch.applications.tinygpt.test_unit_tinygpt_init': ( '20_capstone/capstone.html#test_unit_tinygpt_init',
|
|
'tinytorch/applications/tinygpt.py'),
|
|
'tinytorch.applications.tinygpt.test_unit_training_pipeline': ( '20_capstone/capstone.html#test_unit_training_pipeline',
|
|
'tinytorch/applications/tinygpt.py')},
|
|
'tinytorch.bench': { 'tinytorch.bench.Benchmark': ('19_benchmarking/benchmarking.html#benchmark', 'tinytorch/bench.py'),
|
|
'tinytorch.bench.Benchmark.__init__': ( '19_benchmarking/benchmarking.html#benchmark.__init__',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.Benchmark.compare_models': ( '19_benchmarking/benchmarking.html#benchmark.compare_models',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.Benchmark.run_accuracy_benchmark': ( '19_benchmarking/benchmarking.html#benchmark.run_accuracy_benchmark',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.Benchmark.run_latency_benchmark': ( '19_benchmarking/benchmarking.html#benchmark.run_latency_benchmark',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.Benchmark.run_memory_benchmark': ( '19_benchmarking/benchmarking.html#benchmark.run_memory_benchmark',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.BenchmarkResult': ( '19_benchmarking/benchmarking.html#benchmarkresult',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.BenchmarkResult.__post_init__': ( '19_benchmarking/benchmarking.html#benchmarkresult.__post_init__',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.BenchmarkResult.__str__': ( '19_benchmarking/benchmarking.html#benchmarkresult.__str__',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.BenchmarkResult.to_dict': ( '19_benchmarking/benchmarking.html#benchmarkresult.to_dict',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.BenchmarkSuite': ( '19_benchmarking/benchmarking.html#benchmarksuite',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.BenchmarkSuite.__init__': ( '19_benchmarking/benchmarking.html#benchmarksuite.__init__',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.BenchmarkSuite._estimate_energy_efficiency': ( '19_benchmarking/benchmarking.html#benchmarksuite._estimate_energy_efficiency',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.BenchmarkSuite.generate_report': ( '19_benchmarking/benchmarking.html#benchmarksuite.generate_report',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.BenchmarkSuite.plot_pareto_frontier': ( '19_benchmarking/benchmarking.html#benchmarksuite.plot_pareto_frontier',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.BenchmarkSuite.plot_results': ( '19_benchmarking/benchmarking.html#benchmarksuite.plot_results',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.BenchmarkSuite.run_full_benchmark': ( '19_benchmarking/benchmarking.html#benchmarksuite.run_full_benchmark',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.TinyMLPerf': ('19_benchmarking/benchmarking.html#tinymlperf', 'tinytorch/bench.py'),
|
|
'tinytorch.bench.TinyMLPerf.__init__': ( '19_benchmarking/benchmarking.html#tinymlperf.__init__',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.TinyMLPerf.generate_compliance_report': ( '19_benchmarking/benchmarking.html#tinymlperf.generate_compliance_report',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.TinyMLPerf.run_all_benchmarks': ( '19_benchmarking/benchmarking.html#tinymlperf.run_all_benchmarks',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.TinyMLPerf.run_standard_benchmark': ( '19_benchmarking/benchmarking.html#tinymlperf.run_standard_benchmark',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.test_unit_benchmark': ( '19_benchmarking/benchmarking.html#test_unit_benchmark',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.test_unit_benchmark_result': ( '19_benchmarking/benchmarking.html#test_unit_benchmark_result',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.test_unit_benchmark_suite': ( '19_benchmarking/benchmarking.html#test_unit_benchmark_suite',
|
|
'tinytorch/bench.py'),
|
|
'tinytorch.bench.test_unit_tinymlperf': ( '19_benchmarking/benchmarking.html#test_unit_tinymlperf',
|
|
'tinytorch/bench.py')},
|
|
'tinytorch.benchmarking.benchmark': { 'tinytorch.benchmarking.benchmark.Benchmark': ( '19_benchmarking/benchmarking.html#benchmark',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.Benchmark.__init__': ( '19_benchmarking/benchmarking.html#benchmark.__init__',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.Benchmark.compare_models': ( '19_benchmarking/benchmarking.html#benchmark.compare_models',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.Benchmark.run_accuracy_benchmark': ( '19_benchmarking/benchmarking.html#benchmark.run_accuracy_benchmark',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.Benchmark.run_latency_benchmark': ( '19_benchmarking/benchmarking.html#benchmark.run_latency_benchmark',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.Benchmark.run_memory_benchmark': ( '19_benchmarking/benchmarking.html#benchmark.run_memory_benchmark',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.BenchmarkResult': ( '19_benchmarking/benchmarking.html#benchmarkresult',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.BenchmarkResult.__post_init__': ( '19_benchmarking/benchmarking.html#benchmarkresult.__post_init__',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.BenchmarkResult.__str__': ( '19_benchmarking/benchmarking.html#benchmarkresult.__str__',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.BenchmarkResult.to_dict': ( '19_benchmarking/benchmarking.html#benchmarkresult.to_dict',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.BenchmarkSuite': ( '19_benchmarking/benchmarking.html#benchmarksuite',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.BenchmarkSuite.__init__': ( '19_benchmarking/benchmarking.html#benchmarksuite.__init__',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.BenchmarkSuite._estimate_energy_efficiency': ( '19_benchmarking/benchmarking.html#benchmarksuite._estimate_energy_efficiency',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.BenchmarkSuite.generate_report': ( '19_benchmarking/benchmarking.html#benchmarksuite.generate_report',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.BenchmarkSuite.plot_pareto_frontier': ( '19_benchmarking/benchmarking.html#benchmarksuite.plot_pareto_frontier',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.BenchmarkSuite.plot_results': ( '19_benchmarking/benchmarking.html#benchmarksuite.plot_results',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.BenchmarkSuite.run_full_benchmark': ( '19_benchmarking/benchmarking.html#benchmarksuite.run_full_benchmark',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.TinyMLPerf': ( '19_benchmarking/benchmarking.html#tinymlperf',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.TinyMLPerf.__init__': ( '19_benchmarking/benchmarking.html#tinymlperf.__init__',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.TinyMLPerf.generate_compliance_report': ( '19_benchmarking/benchmarking.html#tinymlperf.generate_compliance_report',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.TinyMLPerf.run_all_benchmarks': ( '19_benchmarking/benchmarking.html#tinymlperf.run_all_benchmarks',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.TinyMLPerf.run_standard_benchmark': ( '19_benchmarking/benchmarking.html#tinymlperf.run_standard_benchmark',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.test_unit_benchmark': ( '19_benchmarking/benchmarking.html#test_unit_benchmark',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.test_unit_benchmark_result': ( '19_benchmarking/benchmarking.html#test_unit_benchmark_result',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.test_unit_benchmark_suite': ( '19_benchmarking/benchmarking.html#test_unit_benchmark_suite',
|
|
'tinytorch/benchmarking/benchmark.py'),
|
|
'tinytorch.benchmarking.benchmark.test_unit_tinymlperf': ( '19_benchmarking/benchmarking.html#test_unit_tinymlperf',
|
|
'tinytorch/benchmarking/benchmark.py')},
|
|
'tinytorch.capstone': { 'tinytorch.capstone.BenchmarkReport': ( '20_capstone/capstone.html#benchmarkreport',
|
|
'tinytorch/capstone.py'),
|
|
'tinytorch.capstone.BenchmarkReport.__init__': ( '20_capstone/capstone.html#benchmarkreport.__init__',
|
|
'tinytorch/capstone.py'),
|
|
'tinytorch.capstone.BenchmarkReport._get_system_info': ( '20_capstone/capstone.html#benchmarkreport._get_system_info',
|
|
'tinytorch/capstone.py'),
|
|
'tinytorch.capstone.BenchmarkReport.benchmark_model': ( '20_capstone/capstone.html#benchmarkreport.benchmark_model',
|
|
'tinytorch/capstone.py'),
|
|
'tinytorch.capstone.SimpleMLP': ('20_capstone/capstone.html#simplemlp', 'tinytorch/capstone.py'),
|
|
'tinytorch.capstone.SimpleMLP.__init__': ( '20_capstone/capstone.html#simplemlp.__init__',
|
|
'tinytorch/capstone.py'),
|
|
'tinytorch.capstone.SimpleMLP.count_parameters': ( '20_capstone/capstone.html#simplemlp.count_parameters',
|
|
'tinytorch/capstone.py'),
|
|
'tinytorch.capstone.SimpleMLP.forward': ( '20_capstone/capstone.html#simplemlp.forward',
|
|
'tinytorch/capstone.py'),
|
|
'tinytorch.capstone.SimpleMLP.parameters': ( '20_capstone/capstone.html#simplemlp.parameters',
|
|
'tinytorch/capstone.py'),
|
|
'tinytorch.capstone.generate_submission': ( '20_capstone/capstone.html#generate_submission',
|
|
'tinytorch/capstone.py'),
|
|
'tinytorch.capstone.save_submission': ( '20_capstone/capstone.html#save_submission',
|
|
'tinytorch/capstone.py')},
|
|
'tinytorch.competition.submit': { 'tinytorch.competition.submit.generate_baseline': ( 'source/20_competition/competition_dev.html#generate_baseline',
|
|
'tinytorch/competition/submit.py'),
|
|
'tinytorch.competition.submit.generate_submission': ( 'source/20_competition/competition_dev.html#generate_submission',
|
|
'tinytorch/competition/submit.py'),
|
|
'tinytorch.competition.submit.load_baseline_model': ( 'source/20_competition/competition_dev.html#load_baseline_model',
|
|
'tinytorch/competition/submit.py'),
|
|
'tinytorch.competition.submit.optimize_for_competition': ( 'source/20_competition/competition_dev.html#optimize_for_competition',
|
|
'tinytorch/competition/submit.py'),
|
|
'tinytorch.competition.submit.validate_installation': ( 'source/20_competition/competition_dev.html#validate_installation',
|
|
'tinytorch/competition/submit.py'),
|
|
'tinytorch.competition.submit.validate_submission': ( 'source/20_competition/competition_dev.html#validate_submission',
|
|
'tinytorch/competition/submit.py'),
|
|
'tinytorch.competition.submit.worked_example_optimization': ( 'source/20_competition/competition_dev.html#worked_example_optimization',
|
|
'tinytorch/competition/submit.py')},
|
|
'tinytorch.core.activations': { 'tinytorch.core.activations.GELU': ( '02_activations/activations.html#gelu',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.GELU.__call__': ( '02_activations/activations.html#gelu.__call__',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.GELU.backward': ( '02_activations/activations.html#gelu.backward',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.GELU.forward': ( '02_activations/activations.html#gelu.forward',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.GELU.parameters': ( '02_activations/activations.html#gelu.parameters',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.ReLU': ( '02_activations/activations.html#relu',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.ReLU.__call__': ( '02_activations/activations.html#relu.__call__',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.ReLU.backward': ( '02_activations/activations.html#relu.backward',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.ReLU.forward': ( '02_activations/activations.html#relu.forward',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.ReLU.parameters': ( '02_activations/activations.html#relu.parameters',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Sigmoid': ( '02_activations/activations.html#sigmoid',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Sigmoid.__call__': ( '02_activations/activations.html#sigmoid.__call__',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Sigmoid.backward': ( '02_activations/activations.html#sigmoid.backward',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Sigmoid.forward': ( '02_activations/activations.html#sigmoid.forward',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Sigmoid.parameters': ( '02_activations/activations.html#sigmoid.parameters',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Softmax': ( '02_activations/activations.html#softmax',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Softmax.__call__': ( '02_activations/activations.html#softmax.__call__',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Softmax.backward': ( '02_activations/activations.html#softmax.backward',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Softmax.forward': ( '02_activations/activations.html#softmax.forward',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Softmax.parameters': ( '02_activations/activations.html#softmax.parameters',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Tanh': ( '02_activations/activations.html#tanh',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Tanh.__call__': ( '02_activations/activations.html#tanh.__call__',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Tanh.backward': ( '02_activations/activations.html#tanh.backward',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Tanh.forward': ( '02_activations/activations.html#tanh.forward',
|
|
'tinytorch/core/activations.py'),
|
|
'tinytorch.core.activations.Tanh.parameters': ( '02_activations/activations.html#tanh.parameters',
|
|
'tinytorch/core/activations.py')},
|
|
'tinytorch.core.attention': { 'tinytorch.core.attention.MultiHeadAttention': ( '12_attention/attention.html#multiheadattention',
|
|
'tinytorch/core/attention.py'),
|
|
'tinytorch.core.attention.MultiHeadAttention.__call__': ( '12_attention/attention.html#multiheadattention.__call__',
|
|
'tinytorch/core/attention.py'),
|
|
'tinytorch.core.attention.MultiHeadAttention.__init__': ( '12_attention/attention.html#multiheadattention.__init__',
|
|
'tinytorch/core/attention.py'),
|
|
'tinytorch.core.attention.MultiHeadAttention.forward': ( '12_attention/attention.html#multiheadattention.forward',
|
|
'tinytorch/core/attention.py'),
|
|
'tinytorch.core.attention.MultiHeadAttention.parameters': ( '12_attention/attention.html#multiheadattention.parameters',
|
|
'tinytorch/core/attention.py'),
|
|
'tinytorch.core.attention.scaled_dot_product_attention': ( '12_attention/attention.html#scaled_dot_product_attention',
|
|
'tinytorch/core/attention.py')},
|
|
'tinytorch.core.autograd': {},
|
|
'tinytorch.core.dataloader': { 'tinytorch.core.dataloader.Compose': ( '08_dataloader/dataloader.html#compose',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.Compose.__call__': ( '08_dataloader/dataloader.html#compose.__call__',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.Compose.__init__': ( '08_dataloader/dataloader.html#compose.__init__',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.DataLoader': ( '08_dataloader/dataloader.html#dataloader',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.DataLoader.__init__': ( '08_dataloader/dataloader.html#dataloader.__init__',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.DataLoader.__iter__': ( '08_dataloader/dataloader.html#dataloader.__iter__',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.DataLoader.__len__': ( '08_dataloader/dataloader.html#dataloader.__len__',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.DataLoader._collate_batch': ( '08_dataloader/dataloader.html#dataloader._collate_batch',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.Dataset': ( '08_dataloader/dataloader.html#dataset',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.Dataset.__getitem__': ( '08_dataloader/dataloader.html#dataset.__getitem__',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.Dataset.__len__': ( '08_dataloader/dataloader.html#dataset.__len__',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.RandomCrop': ( '08_dataloader/dataloader.html#randomcrop',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.RandomCrop.__call__': ( '08_dataloader/dataloader.html#randomcrop.__call__',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.RandomCrop.__init__': ( '08_dataloader/dataloader.html#randomcrop.__init__',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.RandomHorizontalFlip': ( '08_dataloader/dataloader.html#randomhorizontalflip',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.RandomHorizontalFlip.__call__': ( '08_dataloader/dataloader.html#randomhorizontalflip.__call__',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.RandomHorizontalFlip.__init__': ( '08_dataloader/dataloader.html#randomhorizontalflip.__init__',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.TensorDataset': ( '08_dataloader/dataloader.html#tensordataset',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.TensorDataset.__getitem__': ( '08_dataloader/dataloader.html#tensordataset.__getitem__',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.TensorDataset.__init__': ( '08_dataloader/dataloader.html#tensordataset.__init__',
|
|
'tinytorch/core/dataloader.py'),
|
|
'tinytorch.core.dataloader.TensorDataset.__len__': ( '08_dataloader/dataloader.html#tensordataset.__len__',
|
|
'tinytorch/core/dataloader.py')},
|
|
'tinytorch.core.embeddings': { 'tinytorch.core.embeddings.Embedding': ( '11_embeddings/embeddings.html#embedding',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.Embedding.__call__': ( '11_embeddings/embeddings.html#embedding.__call__',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.Embedding.__init__': ( '11_embeddings/embeddings.html#embedding.__init__',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.Embedding.__repr__': ( '11_embeddings/embeddings.html#embedding.__repr__',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.Embedding.forward': ( '11_embeddings/embeddings.html#embedding.forward',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.Embedding.parameters': ( '11_embeddings/embeddings.html#embedding.parameters',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.EmbeddingLayer': ( '11_embeddings/embeddings.html#embeddinglayer',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.EmbeddingLayer.__call__': ( '11_embeddings/embeddings.html#embeddinglayer.__call__',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.EmbeddingLayer.__init__': ( '11_embeddings/embeddings.html#embeddinglayer.__init__',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.EmbeddingLayer.__repr__': ( '11_embeddings/embeddings.html#embeddinglayer.__repr__',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.EmbeddingLayer.forward': ( '11_embeddings/embeddings.html#embeddinglayer.forward',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.EmbeddingLayer.parameters': ( '11_embeddings/embeddings.html#embeddinglayer.parameters',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.PositionalEncoding': ( '11_embeddings/embeddings.html#positionalencoding',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.PositionalEncoding.__call__': ( '11_embeddings/embeddings.html#positionalencoding.__call__',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.PositionalEncoding.__init__': ( '11_embeddings/embeddings.html#positionalencoding.__init__',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.PositionalEncoding.__repr__': ( '11_embeddings/embeddings.html#positionalencoding.__repr__',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.PositionalEncoding.forward': ( '11_embeddings/embeddings.html#positionalencoding.forward',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.PositionalEncoding.parameters': ( '11_embeddings/embeddings.html#positionalencoding.parameters',
|
|
'tinytorch/core/embeddings.py'),
|
|
'tinytorch.core.embeddings.create_sinusoidal_embeddings': ( '11_embeddings/embeddings.html#create_sinusoidal_embeddings',
|
|
'tinytorch/core/embeddings.py')},
|
|
'tinytorch.core.layers': { 'tinytorch.core.layers.Dropout': ('03_layers/layers.html#dropout', 'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Dropout.__call__': ( '03_layers/layers.html#dropout.__call__',
|
|
'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Dropout.__init__': ( '03_layers/layers.html#dropout.__init__',
|
|
'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Dropout.__repr__': ( '03_layers/layers.html#dropout.__repr__',
|
|
'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Dropout.forward': ( '03_layers/layers.html#dropout.forward',
|
|
'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Dropout.parameters': ( '03_layers/layers.html#dropout.parameters',
|
|
'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Layer': ('03_layers/layers.html#layer', 'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Layer.__call__': ( '03_layers/layers.html#layer.__call__',
|
|
'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Layer.__repr__': ( '03_layers/layers.html#layer.__repr__',
|
|
'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Layer.forward': ( '03_layers/layers.html#layer.forward',
|
|
'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Layer.parameters': ( '03_layers/layers.html#layer.parameters',
|
|
'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Linear': ('03_layers/layers.html#linear', 'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Linear.__call__': ( '03_layers/layers.html#linear.__call__',
|
|
'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Linear.__init__': ( '03_layers/layers.html#linear.__init__',
|
|
'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Linear.__repr__': ( '03_layers/layers.html#linear.__repr__',
|
|
'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Linear.forward': ( '03_layers/layers.html#linear.forward',
|
|
'tinytorch/core/layers.py'),
|
|
'tinytorch.core.layers.Linear.parameters': ( '03_layers/layers.html#linear.parameters',
|
|
'tinytorch/core/layers.py')},
|
|
'tinytorch.core.losses': { 'tinytorch.core.losses.BinaryCrossEntropyLoss': ( '04_losses/losses.html#binarycrossentropyloss',
|
|
'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.BinaryCrossEntropyLoss.__call__': ( '04_losses/losses.html#binarycrossentropyloss.__call__',
|
|
'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.BinaryCrossEntropyLoss.__init__': ( '04_losses/losses.html#binarycrossentropyloss.__init__',
|
|
'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.BinaryCrossEntropyLoss.backward': ( '04_losses/losses.html#binarycrossentropyloss.backward',
|
|
'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.BinaryCrossEntropyLoss.forward': ( '04_losses/losses.html#binarycrossentropyloss.forward',
|
|
'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.CrossEntropyLoss': ( '04_losses/losses.html#crossentropyloss',
|
|
'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.CrossEntropyLoss.__call__': ( '04_losses/losses.html#crossentropyloss.__call__',
|
|
'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.CrossEntropyLoss.__init__': ( '04_losses/losses.html#crossentropyloss.__init__',
|
|
'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.CrossEntropyLoss.backward': ( '04_losses/losses.html#crossentropyloss.backward',
|
|
'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.CrossEntropyLoss.forward': ( '04_losses/losses.html#crossentropyloss.forward',
|
|
'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.MSELoss': ('04_losses/losses.html#mseloss', 'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.MSELoss.__call__': ( '04_losses/losses.html#mseloss.__call__',
|
|
'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.MSELoss.__init__': ( '04_losses/losses.html#mseloss.__init__',
|
|
'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.MSELoss.backward': ( '04_losses/losses.html#mseloss.backward',
|
|
'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.MSELoss.forward': ( '04_losses/losses.html#mseloss.forward',
|
|
'tinytorch/core/losses.py'),
|
|
'tinytorch.core.losses.log_softmax': ( '04_losses/losses.html#log_softmax',
|
|
'tinytorch/core/losses.py')},
|
|
'tinytorch.core.optimizers': { 'tinytorch.core.optimizers.Adam': ( '06_optimizers/optimizers.html#adam',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.Adam.__init__': ( '06_optimizers/optimizers.html#adam.__init__',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.Adam.step': ( '06_optimizers/optimizers.html#adam.step',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.AdamW': ( '06_optimizers/optimizers.html#adamw',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.AdamW.__init__': ( '06_optimizers/optimizers.html#adamw.__init__',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.AdamW.step': ( '06_optimizers/optimizers.html#adamw.step',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.Optimizer': ( '06_optimizers/optimizers.html#optimizer',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.Optimizer.__init__': ( '06_optimizers/optimizers.html#optimizer.__init__',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.Optimizer.step': ( '06_optimizers/optimizers.html#optimizer.step',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.Optimizer.zero_grad': ( '06_optimizers/optimizers.html#optimizer.zero_grad',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.SGD': ( '06_optimizers/optimizers.html#sgd',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.SGD.__init__': ( '06_optimizers/optimizers.html#sgd.__init__',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.SGD.get_momentum_state': ( '06_optimizers/optimizers.html#sgd.get_momentum_state',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.SGD.has_momentum': ( '06_optimizers/optimizers.html#sgd.has_momentum',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.SGD.set_momentum_state': ( '06_optimizers/optimizers.html#sgd.set_momentum_state',
|
|
'tinytorch/core/optimizers.py'),
|
|
'tinytorch.core.optimizers.SGD.step': ( '06_optimizers/optimizers.html#sgd.step',
|
|
'tinytorch/core/optimizers.py')},
|
|
'tinytorch.core.spatial': { 'tinytorch.core.spatial.AvgPool2d': ( '09_spatial/spatial.html#avgpool2d',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.AvgPool2d.__call__': ( '09_spatial/spatial.html#avgpool2d.__call__',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.AvgPool2d.__init__': ( '09_spatial/spatial.html#avgpool2d.__init__',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.AvgPool2d.forward': ( '09_spatial/spatial.html#avgpool2d.forward',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.AvgPool2d.parameters': ( '09_spatial/spatial.html#avgpool2d.parameters',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.Conv2d': ('09_spatial/spatial.html#conv2d', 'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.Conv2d.__call__': ( '09_spatial/spatial.html#conv2d.__call__',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.Conv2d.__init__': ( '09_spatial/spatial.html#conv2d.__init__',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.Conv2d.forward': ( '09_spatial/spatial.html#conv2d.forward',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.Conv2d.parameters': ( '09_spatial/spatial.html#conv2d.parameters',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.Conv2dBackward': ( '09_spatial/spatial.html#conv2dbackward',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.Conv2dBackward.__init__': ( '09_spatial/spatial.html#conv2dbackward.__init__',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.Conv2dBackward.apply': ( '09_spatial/spatial.html#conv2dbackward.apply',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.MaxPool2d': ( '09_spatial/spatial.html#maxpool2d',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.MaxPool2d.__call__': ( '09_spatial/spatial.html#maxpool2d.__call__',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.MaxPool2d.__init__': ( '09_spatial/spatial.html#maxpool2d.__init__',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.MaxPool2d.forward': ( '09_spatial/spatial.html#maxpool2d.forward',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.MaxPool2d.parameters': ( '09_spatial/spatial.html#maxpool2d.parameters',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.MaxPool2dBackward': ( '09_spatial/spatial.html#maxpool2dbackward',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.MaxPool2dBackward.__init__': ( '09_spatial/spatial.html#maxpool2dbackward.__init__',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.MaxPool2dBackward.apply': ( '09_spatial/spatial.html#maxpool2dbackward.apply',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.SimpleCNN': ( '09_spatial/spatial.html#simplecnn',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.SimpleCNN.__call__': ( '09_spatial/spatial.html#simplecnn.__call__',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.SimpleCNN.__init__': ( '09_spatial/spatial.html#simplecnn.__init__',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.SimpleCNN.forward': ( '09_spatial/spatial.html#simplecnn.forward',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.SimpleCNN.parameters': ( '09_spatial/spatial.html#simplecnn.parameters',
|
|
'tinytorch/core/spatial.py'),
|
|
'tinytorch.core.spatial.SimpleCNN.relu': ( '09_spatial/spatial.html#simplecnn.relu',
|
|
'tinytorch/core/spatial.py')},
|
|
'tinytorch.core.tensor': { 'tinytorch.core.tensor.Tensor': ('01_tensor/tensor.html#tensor', 'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.__add__': ( '01_tensor/tensor.html#tensor.__add__',
|
|
'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.__getitem__': ( '01_tensor/tensor.html#tensor.__getitem__',
|
|
'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.__init__': ( '01_tensor/tensor.html#tensor.__init__',
|
|
'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.__mul__': ( '01_tensor/tensor.html#tensor.__mul__',
|
|
'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.__repr__': ( '01_tensor/tensor.html#tensor.__repr__',
|
|
'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.__str__': ( '01_tensor/tensor.html#tensor.__str__',
|
|
'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.__sub__': ( '01_tensor/tensor.html#tensor.__sub__',
|
|
'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.__truediv__': ( '01_tensor/tensor.html#tensor.__truediv__',
|
|
'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.backward': ( '01_tensor/tensor.html#tensor.backward',
|
|
'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.matmul': ( '01_tensor/tensor.html#tensor.matmul',
|
|
'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.max': ('01_tensor/tensor.html#tensor.max', 'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.mean': ( '01_tensor/tensor.html#tensor.mean',
|
|
'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.numpy': ( '01_tensor/tensor.html#tensor.numpy',
|
|
'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.reshape': ( '01_tensor/tensor.html#tensor.reshape',
|
|
'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.sum': ('01_tensor/tensor.html#tensor.sum', 'tinytorch/core/tensor.py'),
|
|
'tinytorch.core.tensor.Tensor.transpose': ( '01_tensor/tensor.html#tensor.transpose',
|
|
'tinytorch/core/tensor.py')},
|
|
'tinytorch.core.tokenization': { 'tinytorch.core.tokenization.BPETokenizer': ( '10_tokenization/tokenization.html#bpetokenizer',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.BPETokenizer.__init__': ( '10_tokenization/tokenization.html#bpetokenizer.__init__',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.BPETokenizer._apply_merges': ( '10_tokenization/tokenization.html#bpetokenizer._apply_merges',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.BPETokenizer._build_mappings': ( '10_tokenization/tokenization.html#bpetokenizer._build_mappings',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.BPETokenizer._get_pairs': ( '10_tokenization/tokenization.html#bpetokenizer._get_pairs',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.BPETokenizer._get_word_tokens': ( '10_tokenization/tokenization.html#bpetokenizer._get_word_tokens',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.BPETokenizer.decode': ( '10_tokenization/tokenization.html#bpetokenizer.decode',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.BPETokenizer.encode': ( '10_tokenization/tokenization.html#bpetokenizer.encode',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.BPETokenizer.train': ( '10_tokenization/tokenization.html#bpetokenizer.train',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.CharTokenizer': ( '10_tokenization/tokenization.html#chartokenizer',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.CharTokenizer.__init__': ( '10_tokenization/tokenization.html#chartokenizer.__init__',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.CharTokenizer.build_vocab': ( '10_tokenization/tokenization.html#chartokenizer.build_vocab',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.CharTokenizer.decode': ( '10_tokenization/tokenization.html#chartokenizer.decode',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.CharTokenizer.encode': ( '10_tokenization/tokenization.html#chartokenizer.encode',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.Tokenizer': ( '10_tokenization/tokenization.html#tokenizer',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.Tokenizer.decode': ( '10_tokenization/tokenization.html#tokenizer.decode',
|
|
'tinytorch/core/tokenization.py'),
|
|
'tinytorch.core.tokenization.Tokenizer.encode': ( '10_tokenization/tokenization.html#tokenizer.encode',
|
|
'tinytorch/core/tokenization.py')},
|
|
'tinytorch.core.training': { 'tinytorch.core.training.CosineSchedule': ( '07_training/training.html#cosineschedule',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.CosineSchedule.__init__': ( '07_training/training.html#cosineschedule.__init__',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.CosineSchedule.get_lr': ( '07_training/training.html#cosineschedule.get_lr',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.Trainer': ( '07_training/training.html#trainer',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.Trainer.__init__': ( '07_training/training.html#trainer.__init__',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.Trainer._get_model_state': ( '07_training/training.html#trainer._get_model_state',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.Trainer._get_optimizer_state': ( '07_training/training.html#trainer._get_optimizer_state',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.Trainer._get_scheduler_state': ( '07_training/training.html#trainer._get_scheduler_state',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.Trainer._set_model_state': ( '07_training/training.html#trainer._set_model_state',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.Trainer._set_optimizer_state': ( '07_training/training.html#trainer._set_optimizer_state',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.Trainer._set_scheduler_state': ( '07_training/training.html#trainer._set_scheduler_state',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.Trainer.evaluate': ( '07_training/training.html#trainer.evaluate',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.Trainer.load_checkpoint': ( '07_training/training.html#trainer.load_checkpoint',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.Trainer.save_checkpoint': ( '07_training/training.html#trainer.save_checkpoint',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.Trainer.train_epoch': ( '07_training/training.html#trainer.train_epoch',
|
|
'tinytorch/core/training.py'),
|
|
'tinytorch.core.training.clip_grad_norm': ( '07_training/training.html#clip_grad_norm',
|
|
'tinytorch/core/training.py')},
|
|
'tinytorch.core.transformer': { 'tinytorch.core.transformer.GPT': ( '13_transformers/transformers.html#gpt',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.GPT.__call__': ( '13_transformers/transformers.html#gpt.__call__',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.GPT.__init__': ( '13_transformers/transformers.html#gpt.__init__',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.GPT._create_causal_mask': ( '13_transformers/transformers.html#gpt._create_causal_mask',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.GPT.forward': ( '13_transformers/transformers.html#gpt.forward',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.GPT.generate': ( '13_transformers/transformers.html#gpt.generate',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.GPT.parameters': ( '13_transformers/transformers.html#gpt.parameters',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.LayerNorm': ( '13_transformers/transformers.html#layernorm',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.LayerNorm.__call__': ( '13_transformers/transformers.html#layernorm.__call__',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.LayerNorm.__init__': ( '13_transformers/transformers.html#layernorm.__init__',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.LayerNorm.forward': ( '13_transformers/transformers.html#layernorm.forward',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.LayerNorm.parameters': ( '13_transformers/transformers.html#layernorm.parameters',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.MLP': ( '13_transformers/transformers.html#mlp',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.MLP.__call__': ( '13_transformers/transformers.html#mlp.__call__',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.MLP.__init__': ( '13_transformers/transformers.html#mlp.__init__',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.MLP.forward': ( '13_transformers/transformers.html#mlp.forward',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.MLP.parameters': ( '13_transformers/transformers.html#mlp.parameters',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.TransformerBlock': ( '13_transformers/transformers.html#transformerblock',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.TransformerBlock.__call__': ( '13_transformers/transformers.html#transformerblock.__call__',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.TransformerBlock.__init__': ( '13_transformers/transformers.html#transformerblock.__init__',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.TransformerBlock.forward': ( '13_transformers/transformers.html#transformerblock.forward',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.TransformerBlock.parameters': ( '13_transformers/transformers.html#transformerblock.parameters',
|
|
'tinytorch/core/transformer.py'),
|
|
'tinytorch.core.transformer.create_causal_mask': ( '13_transformers/transformers.html#create_causal_mask',
|
|
'tinytorch/core/transformer.py')},
|
|
'tinytorch.data.loader': { 'tinytorch.data.loader.Compose': ( '08_dataloader/dataloader.html#compose',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.Compose.__call__': ( '08_dataloader/dataloader.html#compose.__call__',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.Compose.__init__': ( '08_dataloader/dataloader.html#compose.__init__',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.DataLoader': ( '08_dataloader/dataloader.html#dataloader',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.DataLoader.__init__': ( '08_dataloader/dataloader.html#dataloader.__init__',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.DataLoader.__iter__': ( '08_dataloader/dataloader.html#dataloader.__iter__',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.DataLoader.__len__': ( '08_dataloader/dataloader.html#dataloader.__len__',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.DataLoader._collate_batch': ( '08_dataloader/dataloader.html#dataloader._collate_batch',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.Dataset': ( '08_dataloader/dataloader.html#dataset',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.Dataset.__getitem__': ( '08_dataloader/dataloader.html#dataset.__getitem__',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.Dataset.__len__': ( '08_dataloader/dataloader.html#dataset.__len__',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.RandomCrop': ( '08_dataloader/dataloader.html#randomcrop',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.RandomCrop.__call__': ( '08_dataloader/dataloader.html#randomcrop.__call__',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.RandomCrop.__init__': ( '08_dataloader/dataloader.html#randomcrop.__init__',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.RandomHorizontalFlip': ( '08_dataloader/dataloader.html#randomhorizontalflip',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.RandomHorizontalFlip.__call__': ( '08_dataloader/dataloader.html#randomhorizontalflip.__call__',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.RandomHorizontalFlip.__init__': ( '08_dataloader/dataloader.html#randomhorizontalflip.__init__',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.TensorDataset': ( '08_dataloader/dataloader.html#tensordataset',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.TensorDataset.__getitem__': ( '08_dataloader/dataloader.html#tensordataset.__getitem__',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.TensorDataset.__init__': ( '08_dataloader/dataloader.html#tensordataset.__init__',
|
|
'tinytorch/data/loader.py'),
|
|
'tinytorch.data.loader.TensorDataset.__len__': ( '08_dataloader/dataloader.html#tensordataset.__len__',
|
|
'tinytorch/data/loader.py')},
|
|
'tinytorch.generation.kv_cache': { 'tinytorch.generation.kv_cache.KVCache': ( '17_memoization/memoization.html#kvcache',
|
|
'tinytorch/generation/kv_cache.py'),
|
|
'tinytorch.generation.kv_cache.KVCache.__init__': ( '17_memoization/memoization.html#kvcache.__init__',
|
|
'tinytorch/generation/kv_cache.py'),
|
|
'tinytorch.generation.kv_cache.KVCache.advance': ( '17_memoization/memoization.html#kvcache.advance',
|
|
'tinytorch/generation/kv_cache.py'),
|
|
'tinytorch.generation.kv_cache.KVCache.get': ( '17_memoization/memoization.html#kvcache.get',
|
|
'tinytorch/generation/kv_cache.py'),
|
|
'tinytorch.generation.kv_cache.KVCache.get_memory_usage': ( '17_memoization/memoization.html#kvcache.get_memory_usage',
|
|
'tinytorch/generation/kv_cache.py'),
|
|
'tinytorch.generation.kv_cache.KVCache.reset': ( '17_memoization/memoization.html#kvcache.reset',
|
|
'tinytorch/generation/kv_cache.py'),
|
|
'tinytorch.generation.kv_cache.KVCache.update': ( '17_memoization/memoization.html#kvcache.update',
|
|
'tinytorch/generation/kv_cache.py'),
|
|
'tinytorch.generation.kv_cache.create_kv_cache': ( '17_memoization/memoization.html#create_kv_cache',
|
|
'tinytorch/generation/kv_cache.py'),
|
|
'tinytorch.generation.kv_cache.disable_kv_cache': ( '17_memoization/memoization.html#disable_kv_cache',
|
|
'tinytorch/generation/kv_cache.py'),
|
|
'tinytorch.generation.kv_cache.enable_kv_cache': ( '17_memoization/memoization.html#enable_kv_cache',
|
|
'tinytorch/generation/kv_cache.py')},
|
|
'tinytorch.models.transformer': { 'tinytorch.models.transformer.GPT': ( 'source/13_transformers/transformers_dev.html#gpt',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.GPT.__init__': ( 'source/13_transformers/transformers_dev.html#gpt.__init__',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.GPT._create_causal_mask': ( 'source/13_transformers/transformers_dev.html#gpt._create_causal_mask',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.GPT.forward': ( 'source/13_transformers/transformers_dev.html#gpt.forward',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.GPT.generate': ( 'source/13_transformers/transformers_dev.html#gpt.generate',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.GPT.parameters': ( 'source/13_transformers/transformers_dev.html#gpt.parameters',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.LayerNorm': ( 'source/13_transformers/transformers_dev.html#layernorm',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.LayerNorm.__call__': ( 'source/13_transformers/transformers_dev.html#layernorm.__call__',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.LayerNorm.__init__': ( 'source/13_transformers/transformers_dev.html#layernorm.__init__',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.LayerNorm.forward': ( 'source/13_transformers/transformers_dev.html#layernorm.forward',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.LayerNorm.parameters': ( 'source/13_transformers/transformers_dev.html#layernorm.parameters',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.MLP': ( 'source/13_transformers/transformers_dev.html#mlp',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.MLP.__call__': ( 'source/13_transformers/transformers_dev.html#mlp.__call__',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.MLP.__init__': ( 'source/13_transformers/transformers_dev.html#mlp.__init__',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.MLP.forward': ( 'source/13_transformers/transformers_dev.html#mlp.forward',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.MLP.parameters': ( 'source/13_transformers/transformers_dev.html#mlp.parameters',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.TransformerBlock': ( 'source/13_transformers/transformers_dev.html#transformerblock',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.TransformerBlock.__call__': ( 'source/13_transformers/transformers_dev.html#transformerblock.__call__',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.TransformerBlock.__init__': ( 'source/13_transformers/transformers_dev.html#transformerblock.__init__',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.TransformerBlock.forward': ( 'source/13_transformers/transformers_dev.html#transformerblock.forward',
|
|
'tinytorch/models/transformer.py'),
|
|
'tinytorch.models.transformer.TransformerBlock.parameters': ( 'source/13_transformers/transformers_dev.html#transformerblock.parameters',
|
|
'tinytorch/models/transformer.py')},
|
|
'tinytorch.optimization.acceleration': { 'tinytorch.optimization.acceleration.fused_gelu': ( '18_acceleration/acceleration.html#fused_gelu',
|
|
'tinytorch/optimization/acceleration.py'),
|
|
'tinytorch.optimization.acceleration.tiled_matmul': ( '18_acceleration/acceleration.html#tiled_matmul',
|
|
'tinytorch/optimization/acceleration.py'),
|
|
'tinytorch.optimization.acceleration.vectorized_matmul': ( '18_acceleration/acceleration.html#vectorized_matmul',
|
|
'tinytorch/optimization/acceleration.py')},
|
|
'tinytorch.optimization.compression': { 'tinytorch.optimization.compression.CompressionComplete': ( '16_compression/compression.html#compressioncomplete',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.CompressionComplete.compress_model': ( '16_compression/compression.html#compressioncomplete.compress_model',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.CompressionComplete.magnitude_prune': ( '16_compression/compression.html#compressioncomplete.magnitude_prune',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.CompressionComplete.measure_sparsity': ( '16_compression/compression.html#compressioncomplete.measure_sparsity',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.CompressionComplete.structured_prune': ( '16_compression/compression.html#compressioncomplete.structured_prune',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.KnowledgeDistillation': ( '16_compression/compression.html#knowledgedistillation',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.KnowledgeDistillation.__init__': ( '16_compression/compression.html#knowledgedistillation.__init__',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.KnowledgeDistillation._cross_entropy': ( '16_compression/compression.html#knowledgedistillation._cross_entropy',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.KnowledgeDistillation._kl_divergence': ( '16_compression/compression.html#knowledgedistillation._kl_divergence',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.KnowledgeDistillation._softmax': ( '16_compression/compression.html#knowledgedistillation._softmax',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.KnowledgeDistillation.distillation_loss': ( '16_compression/compression.html#knowledgedistillation.distillation_loss',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.compress_model': ( '16_compression/compression.html#compress_model',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.low_rank_approximate': ( '16_compression/compression.html#low_rank_approximate',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.magnitude_prune': ( '16_compression/compression.html#magnitude_prune',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.measure_sparsity': ( '16_compression/compression.html#measure_sparsity',
|
|
'tinytorch/optimization/compression.py'),
|
|
'tinytorch.optimization.compression.structured_prune': ( '16_compression/compression.html#structured_prune',
|
|
'tinytorch/optimization/compression.py')},
|
|
'tinytorch.optimization.quantization': { 'tinytorch.optimization.quantization.QuantizationComplete': ( 'source/16_quantization/quantization_dev.html#quantizationcomplete',
|
|
'tinytorch/optimization/quantization.py'),
|
|
'tinytorch.optimization.quantization.QuantizationComplete.compare_models': ( 'source/16_quantization/quantization_dev.html#quantizationcomplete.compare_models',
|
|
'tinytorch/optimization/quantization.py'),
|
|
'tinytorch.optimization.quantization.QuantizationComplete.dequantize_tensor': ( 'source/16_quantization/quantization_dev.html#quantizationcomplete.dequantize_tensor',
|
|
'tinytorch/optimization/quantization.py'),
|
|
'tinytorch.optimization.quantization.QuantizationComplete.quantize_model': ( 'source/16_quantization/quantization_dev.html#quantizationcomplete.quantize_model',
|
|
'tinytorch/optimization/quantization.py'),
|
|
'tinytorch.optimization.quantization.QuantizationComplete.quantize_tensor': ( 'source/16_quantization/quantization_dev.html#quantizationcomplete.quantize_tensor',
|
|
'tinytorch/optimization/quantization.py'),
|
|
'tinytorch.optimization.quantization.dequantize_int8': ( 'source/16_quantization/quantization_dev.html#dequantize_int8',
|
|
'tinytorch/optimization/quantization.py'),
|
|
'tinytorch.optimization.quantization.quantize_int8': ( 'source/16_quantization/quantization_dev.html#quantize_int8',
|
|
'tinytorch/optimization/quantization.py'),
|
|
'tinytorch.optimization.quantization.quantize_model': ( 'source/16_quantization/quantization_dev.html#quantize_model',
|
|
'tinytorch/optimization/quantization.py')},
|
|
'tinytorch.perf.acceleration': { 'tinytorch.perf.acceleration.fused_gelu': ( '18_acceleration/acceleration.html#fused_gelu',
|
|
'tinytorch/perf/acceleration.py'),
|
|
'tinytorch.perf.acceleration.tiled_matmul': ( '18_acceleration/acceleration.html#tiled_matmul',
|
|
'tinytorch/perf/acceleration.py'),
|
|
'tinytorch.perf.acceleration.vectorized_matmul': ( '18_acceleration/acceleration.html#vectorized_matmul',
|
|
'tinytorch/perf/acceleration.py')},
|
|
'tinytorch.perf.compression': { 'tinytorch.perf.compression.Compressor': ( '16_compression/compression.html#compressor',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.Compressor.compress_model': ( '16_compression/compression.html#compressor.compress_model',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.Compressor.magnitude_prune': ( '16_compression/compression.html#compressor.magnitude_prune',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.Compressor.measure_sparsity': ( '16_compression/compression.html#compressor.measure_sparsity',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.Compressor.structured_prune': ( '16_compression/compression.html#compressor.structured_prune',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.KnowledgeDistillation': ( '16_compression/compression.html#knowledgedistillation',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.KnowledgeDistillation.__init__': ( '16_compression/compression.html#knowledgedistillation.__init__',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.KnowledgeDistillation._cross_entropy': ( '16_compression/compression.html#knowledgedistillation._cross_entropy',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.KnowledgeDistillation._kl_divergence': ( '16_compression/compression.html#knowledgedistillation._kl_divergence',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.KnowledgeDistillation._softmax': ( '16_compression/compression.html#knowledgedistillation._softmax',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.KnowledgeDistillation.distillation_loss': ( '16_compression/compression.html#knowledgedistillation.distillation_loss',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.compress_model': ( '16_compression/compression.html#compress_model',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.low_rank_approximate': ( '16_compression/compression.html#low_rank_approximate',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.magnitude_prune': ( '16_compression/compression.html#magnitude_prune',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.measure_sparsity': ( '16_compression/compression.html#measure_sparsity',
|
|
'tinytorch/perf/compression.py'),
|
|
'tinytorch.perf.compression.structured_prune': ( '16_compression/compression.html#structured_prune',
|
|
'tinytorch/perf/compression.py')},
|
|
'tinytorch.perf.memoization': { 'tinytorch.perf.memoization.KVCache': ( '17_memoization/memoization.html#kvcache',
|
|
'tinytorch/perf/memoization.py'),
|
|
'tinytorch.perf.memoization.KVCache.__init__': ( '17_memoization/memoization.html#kvcache.__init__',
|
|
'tinytorch/perf/memoization.py'),
|
|
'tinytorch.perf.memoization.KVCache.advance': ( '17_memoization/memoization.html#kvcache.advance',
|
|
'tinytorch/perf/memoization.py'),
|
|
'tinytorch.perf.memoization.KVCache.get': ( '17_memoization/memoization.html#kvcache.get',
|
|
'tinytorch/perf/memoization.py'),
|
|
'tinytorch.perf.memoization.KVCache.get_memory_usage': ( '17_memoization/memoization.html#kvcache.get_memory_usage',
|
|
'tinytorch/perf/memoization.py'),
|
|
'tinytorch.perf.memoization.KVCache.reset': ( '17_memoization/memoization.html#kvcache.reset',
|
|
'tinytorch/perf/memoization.py'),
|
|
'tinytorch.perf.memoization.KVCache.update': ( '17_memoization/memoization.html#kvcache.update',
|
|
'tinytorch/perf/memoization.py'),
|
|
'tinytorch.perf.memoization.create_kv_cache': ( '17_memoization/memoization.html#create_kv_cache',
|
|
'tinytorch/perf/memoization.py'),
|
|
'tinytorch.perf.memoization.disable_kv_cache': ( '17_memoization/memoization.html#disable_kv_cache',
|
|
'tinytorch/perf/memoization.py'),
|
|
'tinytorch.perf.memoization.enable_kv_cache': ( '17_memoization/memoization.html#enable_kv_cache',
|
|
'tinytorch/perf/memoization.py')},
|
|
'tinytorch.perf.profiling': { 'tinytorch.perf.profiling.Profiler': ( '14_profiling/profiling.html#profiler',
|
|
'tinytorch/perf/profiling.py'),
|
|
'tinytorch.perf.profiling.Profiler.__init__': ( '14_profiling/profiling.html#profiler.__init__',
|
|
'tinytorch/perf/profiling.py'),
|
|
'tinytorch.perf.profiling.Profiler.count_flops': ( '14_profiling/profiling.html#profiler.count_flops',
|
|
'tinytorch/perf/profiling.py'),
|
|
'tinytorch.perf.profiling.Profiler.count_parameters': ( '14_profiling/profiling.html#profiler.count_parameters',
|
|
'tinytorch/perf/profiling.py'),
|
|
'tinytorch.perf.profiling.Profiler.measure_latency': ( '14_profiling/profiling.html#profiler.measure_latency',
|
|
'tinytorch/perf/profiling.py'),
|
|
'tinytorch.perf.profiling.Profiler.measure_memory': ( '14_profiling/profiling.html#profiler.measure_memory',
|
|
'tinytorch/perf/profiling.py'),
|
|
'tinytorch.perf.profiling.Profiler.profile_backward_pass': ( '14_profiling/profiling.html#profiler.profile_backward_pass',
|
|
'tinytorch/perf/profiling.py'),
|
|
'tinytorch.perf.profiling.Profiler.profile_forward_pass': ( '14_profiling/profiling.html#profiler.profile_forward_pass',
|
|
'tinytorch/perf/profiling.py'),
|
|
'tinytorch.perf.profiling.Profiler.profile_layer': ( '14_profiling/profiling.html#profiler.profile_layer',
|
|
'tinytorch/perf/profiling.py'),
|
|
'tinytorch.perf.profiling.analyze_weight_distribution': ( '14_profiling/profiling.html#analyze_weight_distribution',
|
|
'tinytorch/perf/profiling.py'),
|
|
'tinytorch.perf.profiling.quick_profile': ( '14_profiling/profiling.html#quick_profile',
|
|
'tinytorch/perf/profiling.py')},
|
|
'tinytorch.perf.quantization': { 'tinytorch.perf.quantization.QuantizedLinear': ( '15_quantization/quantization.html#quantizedlinear',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.QuantizedLinear.__call__': ( '15_quantization/quantization.html#quantizedlinear.__call__',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.QuantizedLinear.__init__': ( '15_quantization/quantization.html#quantizedlinear.__init__',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.QuantizedLinear.calibrate': ( '15_quantization/quantization.html#quantizedlinear.calibrate',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.QuantizedLinear.forward': ( '15_quantization/quantization.html#quantizedlinear.forward',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.QuantizedLinear.memory_usage': ( '15_quantization/quantization.html#quantizedlinear.memory_usage',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.QuantizedLinear.parameters': ( '15_quantization/quantization.html#quantizedlinear.parameters',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.Quantizer': ( '15_quantization/quantization.html#quantizer',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.Quantizer.compare_models': ( '15_quantization/quantization.html#quantizer.compare_models',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.Quantizer.dequantize_tensor': ( '15_quantization/quantization.html#quantizer.dequantize_tensor',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.Quantizer.quantize_model': ( '15_quantization/quantization.html#quantizer.quantize_model',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.Quantizer.quantize_tensor': ( '15_quantization/quantization.html#quantizer.quantize_tensor',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.compare_model_sizes': ( '15_quantization/quantization.html#compare_model_sizes',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.dequantize_int8': ( '15_quantization/quantization.html#dequantize_int8',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.quantize_int8': ( '15_quantization/quantization.html#quantize_int8',
|
|
'tinytorch/perf/quantization.py'),
|
|
'tinytorch.perf.quantization.quantize_model': ( '15_quantization/quantization.html#quantize_model',
|
|
'tinytorch/perf/quantization.py')},
|
|
'tinytorch.profiling.profiler': { 'tinytorch.profiling.profiler.Profiler': ( '14_profiling/profiling.html#profiler',
|
|
'tinytorch/profiling/profiler.py'),
|
|
'tinytorch.profiling.profiler.Profiler.__init__': ( '14_profiling/profiling.html#profiler.__init__',
|
|
'tinytorch/profiling/profiler.py'),
|
|
'tinytorch.profiling.profiler.Profiler.count_flops': ( '14_profiling/profiling.html#profiler.count_flops',
|
|
'tinytorch/profiling/profiler.py'),
|
|
'tinytorch.profiling.profiler.Profiler.count_parameters': ( '14_profiling/profiling.html#profiler.count_parameters',
|
|
'tinytorch/profiling/profiler.py'),
|
|
'tinytorch.profiling.profiler.Profiler.measure_latency': ( '14_profiling/profiling.html#profiler.measure_latency',
|
|
'tinytorch/profiling/profiler.py'),
|
|
'tinytorch.profiling.profiler.Profiler.measure_memory': ( '14_profiling/profiling.html#profiler.measure_memory',
|
|
'tinytorch/profiling/profiler.py'),
|
|
'tinytorch.profiling.profiler.Profiler.profile_backward_pass': ( '14_profiling/profiling.html#profiler.profile_backward_pass',
|
|
'tinytorch/profiling/profiler.py'),
|
|
'tinytorch.profiling.profiler.Profiler.profile_forward_pass': ( '14_profiling/profiling.html#profiler.profile_forward_pass',
|
|
'tinytorch/profiling/profiler.py'),
|
|
'tinytorch.profiling.profiler.Profiler.profile_layer': ( '14_profiling/profiling.html#profiler.profile_layer',
|
|
'tinytorch/profiling/profiler.py'),
|
|
'tinytorch.profiling.profiler.analyze_weight_distribution': ( '14_profiling/profiling.html#analyze_weight_distribution',
|
|
'tinytorch/profiling/profiler.py'),
|
|
'tinytorch.profiling.profiler.quick_profile': ( '14_profiling/profiling.html#quick_profile',
|
|
'tinytorch/profiling/profiler.py')},
|
|
'tinytorch.text.embeddings': { 'tinytorch.text.embeddings.Embedding': ( '11_embeddings/embeddings.html#embedding',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.Embedding.__call__': ( '11_embeddings/embeddings.html#embedding.__call__',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.Embedding.__init__': ( '11_embeddings/embeddings.html#embedding.__init__',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.Embedding.__repr__': ( '11_embeddings/embeddings.html#embedding.__repr__',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.Embedding.forward': ( '11_embeddings/embeddings.html#embedding.forward',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.Embedding.parameters': ( '11_embeddings/embeddings.html#embedding.parameters',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.EmbeddingLayer': ( '11_embeddings/embeddings.html#embeddinglayer',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.EmbeddingLayer.__call__': ( '11_embeddings/embeddings.html#embeddinglayer.__call__',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.EmbeddingLayer.__init__': ( '11_embeddings/embeddings.html#embeddinglayer.__init__',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.EmbeddingLayer.__repr__': ( '11_embeddings/embeddings.html#embeddinglayer.__repr__',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.EmbeddingLayer.forward': ( '11_embeddings/embeddings.html#embeddinglayer.forward',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.EmbeddingLayer.parameters': ( '11_embeddings/embeddings.html#embeddinglayer.parameters',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.PositionalEncoding': ( '11_embeddings/embeddings.html#positionalencoding',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.PositionalEncoding.__call__': ( '11_embeddings/embeddings.html#positionalencoding.__call__',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.PositionalEncoding.__init__': ( '11_embeddings/embeddings.html#positionalencoding.__init__',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.PositionalEncoding.__repr__': ( '11_embeddings/embeddings.html#positionalencoding.__repr__',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.PositionalEncoding.forward': ( '11_embeddings/embeddings.html#positionalencoding.forward',
|
|
'tinytorch/text/embeddings.py'),
|
|
'tinytorch.text.embeddings.PositionalEncoding.parameters': ( '11_embeddings/embeddings.html#positionalencoding.parameters',
|
|
'tinytorch/text/embeddings.py')},
|
|
'tinytorch.text.tokenization': { 'tinytorch.text.tokenization.BPETokenizer': ( '10_tokenization/tokenization.html#bpetokenizer',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.BPETokenizer.__init__': ( '10_tokenization/tokenization.html#bpetokenizer.__init__',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.BPETokenizer._apply_merges': ( '10_tokenization/tokenization.html#bpetokenizer._apply_merges',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.BPETokenizer._build_mappings': ( '10_tokenization/tokenization.html#bpetokenizer._build_mappings',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.BPETokenizer._get_pairs': ( '10_tokenization/tokenization.html#bpetokenizer._get_pairs',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.BPETokenizer._get_word_tokens': ( '10_tokenization/tokenization.html#bpetokenizer._get_word_tokens',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.BPETokenizer.decode': ( '10_tokenization/tokenization.html#bpetokenizer.decode',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.BPETokenizer.encode': ( '10_tokenization/tokenization.html#bpetokenizer.encode',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.BPETokenizer.train': ( '10_tokenization/tokenization.html#bpetokenizer.train',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.CharTokenizer': ( '10_tokenization/tokenization.html#chartokenizer',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.CharTokenizer.__init__': ( '10_tokenization/tokenization.html#chartokenizer.__init__',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.CharTokenizer.build_vocab': ( '10_tokenization/tokenization.html#chartokenizer.build_vocab',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.CharTokenizer.decode': ( '10_tokenization/tokenization.html#chartokenizer.decode',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.CharTokenizer.encode': ( '10_tokenization/tokenization.html#chartokenizer.encode',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.Tokenizer': ( '10_tokenization/tokenization.html#tokenizer',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.Tokenizer.decode': ( '10_tokenization/tokenization.html#tokenizer.decode',
|
|
'tinytorch/text/tokenization.py'),
|
|
'tinytorch.text.tokenization.Tokenizer.encode': ( '10_tokenization/tokenization.html#tokenizer.encode',
|
|
'tinytorch/text/tokenization.py')}}}
|