OptionBuilder.OptionBuilder Class Reference

Public Member Functions
	__init__ (self, one_cmd_type)
	build (self, commands)

Data Fields
	type

Protected Member Functions
	_build_default (self, commands)
	_build_with_unknown_command (self, commands)
	_build_import (self, commands)
	_build_optimize (self, commands)
	_build_quantize (self, commands)

Protected Attributes
	_build_default

Detailed Description

Definition at line 20 of file OptionBuilder.py.

Constructor & Destructor Documentation

__init__()

OptionBuilder.OptionBuilder.__init__	(		self,
			one_cmd_type
	)

Definition at line 21 of file OptionBuilder.py.

    def __init__(self, one_cmd_type):
        self.type = one_cmd_type
 

Member Function Documentation

_build_default()

OptionBuilder.OptionBuilder._build_default (   self,   commands  )

protected

Definition at line 24 of file OptionBuilder.py.

    def _build_default(self, commands):
        options = []
        for k, v in commands.items():
            options.extend(['--' + k, v])
        return options
 

_build_import()

OptionBuilder.OptionBuilder._build_import (   self,   commands  )

protected

Definition at line 41 of file OptionBuilder.py.

    def _build_import(self, commands):
        options = []
        arg_0 = ['save_intermediate']
        for k, v in commands.items():
            if k in arg_0 and v == "True":
                options.extend(['--' + k])
                continue
            options.extend(['--' + k, v])
        return options
 

_build_optimize()

OptionBuilder.OptionBuilder._build_optimize (   self,   commands  )

protected

Definition at line 51 of file OptionBuilder.py.

    def _build_optimize(self, commands):
        options = []
        arg_0 = ['generate_profile_data']
        arg_1 = ['input_path', 'output_path', 'change_outputs']
        for k, v in commands.items():
            if k in arg_1:
                options.extend(['--' + k, v])
                continue
            if k in arg_0 and v == 'True':
                options.extend(['--' + k])
                continue
            for opt in CONSTANT.OPTIMIZATION_OPTS:
                if k == opt[0] and v == "True":
                    options.extend(['--' + k])
                    break
        return options
 

_build_quantize()

OptionBuilder.OptionBuilder._build_quantize (   self,   commands  )

protected

Definition at line 68 of file OptionBuilder.py.

    def _build_quantize(self, commands):
        options = []
        arg_0 = [
            'generate_profile_data', 'save_intermediate', 'TF-style_maxpool',
            'evaluate_result', 'print_mae', 'print_mape', 'print_mpeir',
            'print_top1_match', 'print_top5_match', 'force_quantparam', 'copy_quantparam'
        ]
        for k, v in commands.items():
            if k in arg_0 and v == "True":
                options.extend(['--' + k])
                continue
            options.extend(['--' + k, v])
        return options
 

_build_with_unknown_command()

OptionBuilder.OptionBuilder._build_with_unknown_command (   self,   commands  )

protected

Definition at line 30 of file OptionBuilder.py.

    def _build_with_unknown_command(self, commands):
        COMMAND_K = 'command'
        options = []
        for k, v in commands.items():
            if k == COMMAND_K:
                continue
            options.extend(['--' + k, v])
        options.extend(['--'])
        options.extend(commands[COMMAND_K].split())
        return options
 

build()

OptionBuilder.OptionBuilder.build	(		self,
			commands
	)

Definition at line 82 of file OptionBuilder.py.

    def build(self, commands):
        cmd_book = dict.fromkeys(
            ['one-import-bcq', 'one-import-tflite', 'one-pack', 'one-partition'],
            self._build_default)
        cmd_book['one-codegen'] = self._build_with_unknown_command
        cmd_book['one-import-onnx'] = self._build_import
        cmd_book['one-import-pytorch'] = self._build_import
        cmd_book['one-import-tf'] = self._build_import
        cmd_book['one-infer'] = self._build_with_unknown_command
        cmd_book['one-optimize'] = self._build_optimize
        cmd_book['one-profile'] = self._build_with_unknown_command
        cmd_book['one-quantize'] = self._build_quantize
 
        return cmd_book[self.type](commands)

Field Documentation

_build_default

OptionBuilder.OptionBuilder._build_default

protected

Definition at line 85 of file OptionBuilder.py.

type

OptionBuilder.OptionBuilder.type

Definition at line 22 of file OptionBuilder.py.

---

The documentation for this class was generated from the following file:

• compiler/one-cmds/onelib/OptionBuilder.py