def __init__(self, input_graph, mode): """Sets up the class to rewrite a float graph. Args: input_graph: A float graph to transform. mode: A string controlling how quantization is performed - round, quantize, eightbit, or weights. Raises: ValueError: Two nodes with the same name were found in the graph. """ self.input_graph = input_graph self.nodes_map = self.create_nodes_map(input_graph) self.output_graph = None self.mode = mode load_quantized_ops_so.Load() load_quantized_kernels_so.Load()
def __init__(self, method_name="runTest"): super(DequantizeOpTest, self).__init__(method_name) load_quantized_ops_so.Load() load_quantized_kernels_so.Load()