diff --git a/tinytorch/_modidx.py b/tinytorch/_modidx.py index c7e9e588..3cd9a0a8 100644 --- a/tinytorch/_modidx.py +++ b/tinytorch/_modidx.py @@ -344,23 +344,23 @@ d = { 'settings': { 'branch': 'main', 'tinytorch/data/loader.py'), 'tinytorch.data.loader.TensorDataset.__len__': ( '08_dataloader/dataloader_dev.html#tensordataset.__len__', 'tinytorch/data/loader.py')}, - 'tinytorch.generation.kv_cache': { 'tinytorch.generation.kv_cache.KVCache': ( '14_kvcaching/kvcaching_dev.html#kvcache', + 'tinytorch.generation.kv_cache': { 'tinytorch.generation.kv_cache.KVCache': ( '15_memoization/memoization_dev.html#kvcache', 'tinytorch/generation/kv_cache.py'), - 'tinytorch.generation.kv_cache.KVCache.__init__': ( '14_kvcaching/kvcaching_dev.html#kvcache.__init__', + 'tinytorch.generation.kv_cache.KVCache.__init__': ( '15_memoization/memoization_dev.html#kvcache.__init__', 'tinytorch/generation/kv_cache.py'), - 'tinytorch.generation.kv_cache.KVCache.advance': ( '14_kvcaching/kvcaching_dev.html#kvcache.advance', + 'tinytorch.generation.kv_cache.KVCache.advance': ( '15_memoization/memoization_dev.html#kvcache.advance', 'tinytorch/generation/kv_cache.py'), - 'tinytorch.generation.kv_cache.KVCache.get': ( '14_kvcaching/kvcaching_dev.html#kvcache.get', + 'tinytorch.generation.kv_cache.KVCache.get': ( '15_memoization/memoization_dev.html#kvcache.get', 'tinytorch/generation/kv_cache.py'), - 'tinytorch.generation.kv_cache.KVCache.get_memory_usage': ( '14_kvcaching/kvcaching_dev.html#kvcache.get_memory_usage', + 'tinytorch.generation.kv_cache.KVCache.get_memory_usage': ( '15_memoization/memoization_dev.html#kvcache.get_memory_usage', 'tinytorch/generation/kv_cache.py'), - 'tinytorch.generation.kv_cache.KVCache.reset': ( '14_kvcaching/kvcaching_dev.html#kvcache.reset', + 'tinytorch.generation.kv_cache.KVCache.reset': ( '15_memoization/memoization_dev.html#kvcache.reset', 'tinytorch/generation/kv_cache.py'), - 'tinytorch.generation.kv_cache.KVCache.update': ( '14_kvcaching/kvcaching_dev.html#kvcache.update', + 'tinytorch.generation.kv_cache.KVCache.update': ( '15_memoization/memoization_dev.html#kvcache.update', 'tinytorch/generation/kv_cache.py'), - 'tinytorch.generation.kv_cache.disable_kv_cache': ( '14_kvcaching/kvcaching_dev.html#disable_kv_cache', + 'tinytorch.generation.kv_cache.disable_kv_cache': ( '15_memoization/memoization_dev.html#disable_kv_cache', 'tinytorch/generation/kv_cache.py'), - 'tinytorch.generation.kv_cache.enable_kv_cache': ( '14_kvcaching/kvcaching_dev.html#enable_kv_cache', + 'tinytorch.generation.kv_cache.enable_kv_cache': ( '15_memoization/memoization_dev.html#enable_kv_cache', 'tinytorch/generation/kv_cache.py')}, 'tinytorch.models.transformer': { 'tinytorch.models.transformer.GPT': ( '13_transformers/transformers_dev.html#gpt', 'tinytorch/models/transformer.py'), @@ -418,22 +418,28 @@ d = { 'settings': { 'branch': 'main', 'tinytorch/optimization/quantization.py'), 'tinytorch.optimization.quantization.quantize_model': ( '17_quantization/quantization_dev.html#quantize_model', 'tinytorch/optimization/quantization.py')}, - 'tinytorch.profiling.profiler': { 'tinytorch.profiling.profiler.Profiler': ( '15_profiling/profiling_dev.html#profiler', + 'tinytorch.profiling.profiler': { 'tinytorch.profiling.profiler.Profiler': ( '14_profiling/profiling_dev.html#profiler', 'tinytorch/profiling/profiler.py'), - 'tinytorch.profiling.profiler.Profiler.__init__': ( '15_profiling/profiling_dev.html#profiler.__init__', + 'tinytorch.profiling.profiler.Profiler.__init__': ( '14_profiling/profiling_dev.html#profiler.__init__', 'tinytorch/profiling/profiler.py'), - 'tinytorch.profiling.profiler.ProfilerComplete': ( '15_profiling/profiling_dev.html#profilercomplete', - 'tinytorch/profiling/profiler.py'), - 'tinytorch.profiling.profiler.ProfilerComplete.__init__': ( '15_profiling/profiling_dev.html#profilercomplete.__init__', + 'tinytorch.profiling.profiler.Profiler.count_flops': ( '14_profiling/profiling_dev.html#profiler.count_flops', + 'tinytorch/profiling/profiler.py'), + 'tinytorch.profiling.profiler.Profiler.count_parameters': ( '14_profiling/profiling_dev.html#profiler.count_parameters', 'tinytorch/profiling/profiler.py'), - 'tinytorch.profiling.profiler.ProfilerComplete.count_flops': ( '15_profiling/profiling_dev.html#profilercomplete.count_flops', - 'tinytorch/profiling/profiler.py'), - 'tinytorch.profiling.profiler.ProfilerComplete.count_parameters': ( '15_profiling/profiling_dev.html#profilercomplete.count_parameters', - 'tinytorch/profiling/profiler.py'), - 'tinytorch.profiling.profiler.ProfilerComplete.measure_latency': ( '15_profiling/profiling_dev.html#profilercomplete.measure_latency', - 'tinytorch/profiling/profiler.py'), - 'tinytorch.profiling.profiler.ProfilerComplete.measure_memory': ( '15_profiling/profiling_dev.html#profilercomplete.measure_memory', - 'tinytorch/profiling/profiler.py')}, + 'tinytorch.profiling.profiler.Profiler.measure_latency': ( '14_profiling/profiling_dev.html#profiler.measure_latency', + 'tinytorch/profiling/profiler.py'), + 'tinytorch.profiling.profiler.Profiler.measure_memory': ( '14_profiling/profiling_dev.html#profiler.measure_memory', + 'tinytorch/profiling/profiler.py'), + 'tinytorch.profiling.profiler.Profiler.profile_backward_pass': ( '14_profiling/profiling_dev.html#profiler.profile_backward_pass', + 'tinytorch/profiling/profiler.py'), + 'tinytorch.profiling.profiler.Profiler.profile_forward_pass': ( '14_profiling/profiling_dev.html#profiler.profile_forward_pass', + 'tinytorch/profiling/profiler.py'), + 'tinytorch.profiling.profiler.Profiler.profile_layer': ( '14_profiling/profiling_dev.html#profiler.profile_layer', + 'tinytorch/profiling/profiler.py'), + 'tinytorch.profiling.profiler.analyze_weight_distribution': ( '14_profiling/profiling_dev.html#analyze_weight_distribution', + 'tinytorch/profiling/profiler.py'), + 'tinytorch.profiling.profiler.quick_profile': ( '14_profiling/profiling_dev.html#quick_profile', + 'tinytorch/profiling/profiler.py')}, 'tinytorch.text.embeddings': { 'tinytorch.text.embeddings.Embedding': ( '11_embeddings/embeddings_dev.html#embedding', 'tinytorch/text/embeddings.py'), 'tinytorch.text.embeddings.Embedding.__init__': ( '11_embeddings/embeddings_dev.html#embedding.__init__',