Index A | B | C | D | E | F | H | M | P | R | S A after_forward() (experiments.util.measurements.MemoryMeasurement method) B batch_normMemSave() (in module memsave_torch.nn.functional) C clear_file() (experiments.util.collect_results.ResultsCollector method) collect_from_file() (experiments.util.collect_results.ResultsCollector method) conv_model_fns (in module experiments.util.models) convert_to_memory_saving() (in module memsave_torch.nn) convMemSave() (in module memsave_torch.nn.functional) D dropoutMemSave() (in module memsave_torch.nn.functional) E estimate_mem_savings() (in module experiments.util.estimate) estimate_speedup() (in module experiments.util.estimate) experiments.util module experiments.util.collect_results module experiments.util.estimate module experiments.util.measurements module experiments.util.models module F finish() (experiments.util.collect_results.ResultsCollector method) forward() (memsave_torch.nn.MemSaveBatchNorm2d method) (memsave_torch.nn.MemSaveConv1d method) (memsave_torch.nn.MemSaveConv2d method) (memsave_torch.nn.MemSaveConv3d method) (memsave_torch.nn.MemSaveConvTranspose1d method) (memsave_torch.nn.MemSaveConvTranspose2d method) (memsave_torch.nn.MemSaveConvTranspose3d method) (memsave_torch.nn.MemSaveMaxPool2d method) (memsave_torch.nn.MemSaveReLU method) forward_backward() (experiments.util.measurements.RuntimeMeasurement method) from_nn_BatchNorm2d() (memsave_torch.nn.MemSaveBatchNorm2d class method) from_nn_Conv1d() (memsave_torch.nn.MemSaveConv1d class method) from_nn_Conv2d() (memsave_torch.nn.MemSaveConv2d class method) from_nn_Conv3d() (memsave_torch.nn.MemSaveConv3d class method) from_nn_ConvTranspose1d() (memsave_torch.nn.MemSaveConvTranspose1d class method) from_nn_ConvTranspose2d() (memsave_torch.nn.MemSaveConvTranspose2d class method) from_nn_ConvTranspose3d() (memsave_torch.nn.MemSaveConvTranspose3d class method) from_nn_MaxPool2d() (memsave_torch.nn.MemSaveMaxPool2d class method) from_nn_ReLU() (memsave_torch.nn.MemSaveReLU class method) H hf_transformers_models_map (in module experiments.util.models) hyperparam_str() (in module experiments.util.collect_results) M make_case_str() (in module experiments.util.collect_results) maxpool2dMemSave() (in module memsave_torch.nn.functional) maybe_synchronize() (in module experiments.util.measurements) Measurement (class in experiments.util.measurements) MemoryMeasurement (class in experiments.util.measurements) memsave_torch module memsave_torch.nn module memsave_torch.nn.functional module MemSaveBatchNorm2d (class in memsave_torch.nn) MemSaveConv1d (class in memsave_torch.nn) MemSaveConv2d (class in memsave_torch.nn) MemSaveConv3d (class in memsave_torch.nn) MemSaveConvTranspose1d (class in memsave_torch.nn) MemSaveConvTranspose2d (class in memsave_torch.nn) MemSaveConvTranspose3d (class in memsave_torch.nn) MemSaveMaxPool2d (class in memsave_torch.nn) MemSaveReLU (class in memsave_torch.nn) module experiments.util experiments.util.collect_results experiments.util.estimate experiments.util.measurements experiments.util.models memsave_torch memsave_torch.nn memsave_torch.nn.functional P parse_case() (in module experiments.util.estimate) prefix_in_pairs() (in module experiments.util.models) R reluMemSave() (in module memsave_torch.nn.functional) ResultsCollector (class in experiments.util.collect_results) RuntimeMeasurement (class in experiments.util.measurements) S select_cases() (in module experiments.util.collect_results) separate_grad_arguments() (in module experiments.util.measurements) separate_surgical() (in module experiments.util.measurements) set_up() (experiments.util.measurements.Measurement method) skip_case_check() (in module experiments.util.estimate)