SYMBOL INDEX (53 symbols across 7 files) FILE: build_helper.py function print_warning (line 9) | def print_warning(*lines): function get_path (line 16) | def get_path(key): function search_on_path (line 20) | def search_on_path(filenames): function get_compiler_setting (line 34) | def get_compiler_setting(): function check_cuda_version (line 78) | def check_cuda_version(): function check_cudnn_version (line 109) | def check_cudnn_version(): function build_and_run (line 137) | def build_and_run(compiler, FILE: nnieqat/quantize.py class GFPQParamSt (line 17) | class GFPQParamSt(ctypes.Structure): class _types (line 22) | class _types: class QuantAndDeQuantGPU (line 28) | class QuantAndDeQuantGPU(): method __init__ (line 31) | def __init__(self, method __call__ (line 50) | def __call__(self, tensor, mode=0): function _fuse_conv_bn_weights (line 117) | def _fuse_conv_bn_weights(conv_w, conv_b, bn_rm, bn_rv, bn_eps, bn_w, bn... function _fuse_conv_bn (line 145) | def _fuse_conv_bn(conv, bn): function _fuse_modules (line 152) | def _fuse_modules(model): function freeze_bn (line 200) | def freeze_bn(m, freeze_bn_affine=True): function merge_freeze_bn (line 221) | def merge_freeze_bn(model): function unquant_weight (line 235) | def unquant_weight(m): function quant_dequant_weight (line 250) | def quant_dequant_weight(m): function _quantizing_activation (line 271) | def _quantizing_activation(module, input, output): function _quantizing_data (line 291) | def _quantizing_data(module, input): function _quantizing_weight (line 313) | def _quantizing_weight(module, input): function register_quantization_hook (line 326) | def register_quantization_hook(model, function test (line 368) | def test(): FILE: src/fake_quantize.cpp function Tensor (line 7) | Tensor fake_quantize(Tensor a, int bit_width){ function PYBIND11_MODULE (line 12) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m){ FILE: tests/test_cifar10.py class Net (line 14) | class Net(nn.Module): method __init__ (line 15) | def __init__(self): method forward (line 24) | def forward(self, x): class TestCifar10 (line 33) | class TestCifar10(unittest.TestCase): method test (line 34) | def test(self): FILE: tests/test_imagenet.py function main (line 81) | def main(): function main_worker (line 116) | def main_worker(gpu, ngpus_per_node, args): function train (line 271) | def train(train_loader, model, criterion, optimizer, epoch, args): function validate (line 320) | def validate(val_loader, model, criterion, args): function save_checkpoint (line 365) | def save_checkpoint(state, is_best, filename='checkpoint.pth.tar'): class AverageMeter (line 371) | class AverageMeter(object): method __init__ (line 373) | def __init__(self, name, fmt=':f'): method reset (line 378) | def reset(self): method update (line 384) | def update(self, val, n=1): method __str__ (line 390) | def __str__(self): class ProgressMeter (line 395) | class ProgressMeter(object): method __init__ (line 396) | def __init__(self, num_batches, meters, prefix=""): method display (line 401) | def display(self, batch): method _get_batch_fmtstr (line 406) | def _get_batch_fmtstr(self, num_batches): function adjust_learning_rate (line 412) | def adjust_learning_rate(optimizer, epoch, args): function accuracy (line 419) | def accuracy(output, target, topk=(1,)): FILE: tests/test_merge_freeze_bn.py class TestMergeFreezeBNImpl (line 10) | class TestMergeFreezeBNImpl(unittest.TestCase): method conv_bn (line 11) | def conv_bn(inp, method conv_1x1_bn (line 19) | def conv_1x1_bn(inp, oup, conv_layer=nn.Conv2d, norm_layer=nn.BatchNor... method test (line 27) | def test(self, m): FILE: tests/test_quant_impl.py class TestQuantImpl (line 15) | class TestQuantImpl(unittest.TestCase): method test (line 43) | def test(self, data):