Python Load Examples

Example #1

File: quantize_graph.py Project: michaelasapp/python-mcparseface

    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()

Example #2

File: dequantize_op_test.py Project: michaelasapp/python-mcparseface

 def __init__(self, method_name="runTest"):
     super(DequantizeOpTest, self).__init__(method_name)
     load_quantized_ops_so.Load()
     load_quantized_kernels_so.Load()