mmdnn/conversion/onnx/onnx_emitter.py [228:265]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        if IR_node.get_attr('scale'):
            self.add_body(1, "{:15} = _weights_dict['{}']['scale'].squeeze()".format(
                IR_node.variable_name + '_scale_array',
                IR_node.name))
        else:
            self.add_body(1, "{:15} = np.ndarray(_weights_dict['{}']['bias'].shape, dtype=_weights_dict['{}']['bias'].dtype).squeeze()".format(
                              IR_node.variable_name + '_scale_array',
                              IR_node.name,
                              IR_node.name))
            self.add_body(1, "{:15}.fill(1)".format(IR_node.variable_name + '_scale_array'))

        self.add_body(1, "{:15} = helper.make_tensor_value_info('{}', onnx.mapping.NP_TYPE_TO_TENSOR_TYPE[{}.dtype], list({}.shape))".format(
            IR_node.variable_name + '_scale',
            IR_node.variable_name + '_scale',
            IR_node.variable_name + '_scale_array',
            IR_node.variable_name + '_scale_array'))
    
        self.add_body(1, "{:15} = helper.make_tensor(name='{}', data_type=onnx.mapping.NP_TYPE_TO_TENSOR_TYPE[{}.dtype], dims={}.shape, vals={})".format(
                          IR_node.variable_name + '_scale_init',
                          IR_node.variable_name + '_scale',
                          IR_node.variable_name + '_scale_array',
                          IR_node.variable_name + '_scale_array',
                          IR_node.variable_name + '_scale_array'))
        self.add_body(1, "{:15} = _weights_dict['{}']['bias'].squeeze()".format(
            IR_node.variable_name + '_bias_array',
            IR_node.name))
        self.add_body(1, "{:15} = helper.make_tensor_value_info('{}', onnx.mapping.NP_TYPE_TO_TENSOR_TYPE[{}.dtype], list({}.shape))".format(
                        IR_node.variable_name + '_bias',
                        IR_node.variable_name + '_bias',
                        IR_node.variable_name + '_bias_array',
                        IR_node.variable_name + '_bias_array'))

        self.add_body(1, "{:15} = helper.make_tensor(name='{}', data_type=onnx.mapping.NP_TYPE_TO_TENSOR_TYPE[{}.dtype], dims={}.shape, vals={}.flatten().astype(float))".format(
                            IR_node.variable_name + '_bias_init',
                            IR_node.variable_name + '_bias',
                            IR_node.variable_name + '_bias_array',
                            IR_node.variable_name + '_bias_array',
                            IR_node.variable_name + '_bias_array'))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



mmdnn/conversion/onnx/onnx_emitter.py [325:361]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        if IR_node.get_attr('scale'):
            self.add_body(1, "{:15} = _weights_dict['{}']['scale'].squeeze()".format(
                IR_node.variable_name + '_scale_array',
                IR_node.name))
        else:
            self.add_body(1, "{:15} = np.ndarray(_weights_dict['{}']['bias'].shape, dtype=_weights_dict['{}']['bias'].dtype).squeeze()".format(
                              IR_node.variable_name + '_scale_array',
                              IR_node.name,
                              IR_node.name))
            self.add_body(1, "{:15}.fill(1)".format(IR_node.variable_name + '_scale_array'))
        self.add_body(1, "{:15} = helper.make_tensor_value_info('{}', onnx.mapping.NP_TYPE_TO_TENSOR_TYPE[{}.dtype], list({}.shape))".format(
            IR_node.variable_name + '_scale',
            IR_node.variable_name + '_scale',
            IR_node.variable_name + '_scale_array',
            IR_node.variable_name + '_scale_array'))
    
        self.add_body(1, "{:15} = helper.make_tensor(name='{}', data_type=onnx.mapping.NP_TYPE_TO_TENSOR_TYPE[{}.dtype], dims={}.shape, vals={})".format(
                            IR_node.variable_name + '_scale_init',
                            IR_node.variable_name + '_scale',
                            IR_node.variable_name + '_scale_array',
                            IR_node.variable_name + '_scale_array',
                            IR_node.variable_name + '_scale_array'))
        self.add_body(1, "{:15} = _weights_dict['{}']['bias'].squeeze()".format(
            IR_node.variable_name + '_bias_array',
            IR_node.name))
        self.add_body(1, "{:15} = helper.make_tensor_value_info('{}', onnx.mapping.NP_TYPE_TO_TENSOR_TYPE[{}.dtype], list({}.shape))".format(
            IR_node.variable_name + '_bias',
            IR_node.variable_name + '_bias',
            IR_node.variable_name + '_bias_array',
            IR_node.variable_name + '_bias_array'))

        self.add_body(1, "{:15} = helper.make_tensor(name='{}', data_type=onnx.mapping.NP_TYPE_TO_TENSOR_TYPE[{}.dtype], dims={}.shape, vals={}.flatten().astype(float))".format(
                            IR_node.variable_name + '_bias_init',
                            IR_node.variable_name + '_bias',
                            IR_node.variable_name + '_bias_array',
                            IR_node.variable_name + '_bias_array',
                            IR_node.variable_name + '_bias_array'))
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -