Searched refs:quantize_rnn_modules (Results 1 – 2 of 2) sorted by relevance
104 cell_int8 = torch.jit.quantized.quantize_rnn_modules(111 cell_fp16 = torch.jit.quantized.quantize_rnn_modules(
96 def quantize_rnn_modules(module, dtype=torch.int8): function