Ejemplos de CxxConfig.CxxConfig en Python

Ejemplo n.º 1

 def sample_predictor_configs(self):
     config = CxxConfig()
     config.set_valid_places({
         Place(TargetType.X86, PrecisionType.FP32, DataLayoutType.NCHW),
         Place(TargetType.X86, PrecisionType.INT64, DataLayoutType.NCHW)
     })
     yield config, ["sequence_concat"], (1e-5, 1e-5)

Ejemplo n.º 2

    def sample_predictor_configs(self):
        config = CxxConfig()
        config.set_valid_places(
            {Place(TargetType.ARM, PrecisionType.FP32, DataLayoutType.NCHW)})
        config.set_threads(1)

        yield config, ["fusion_elementwise_add_activation"], (1e-5, 1e-5)

Ejemplo n.º 3

 def sample_predictor_configs(self):
     config = CxxConfig()
     config.set_valid_places({
         Place(TargetType.Host, PrecisionType.INT64, DataLayoutType.NCHW),
         Place(TargetType.Host, PrecisionType.INT32, DataLayoutType.NCHW)
     })
     yield config, ["ctc_align"], (1e-5, 1e-5)

Ejemplo n.º 4

 def target_to_predictor_configs(self, target: str) -> List[CxxConfig]:
     configs_ = []
     for elem_ in self.valid_places:
         if target in elem_[0]:
             for thread_ in self.thread_num:
                 config_ = CxxConfig()
                 config_.set_valid_places(elem_)
                 config_.set_threads(thread_)
                 configs_.append(config_)
     return configs_

Ejemplo n.º 5

Archivo: auto_scan_base.py Proyecto: newway/Paddle-Lite

 def target_to_predictor_configs(self, target: str) -> List[CxxConfig]:
     configs_ = []
     for elem_ in self.valid_places:
         if target in elem_[0]:
             for thread_ in self.thread_num:
                 config_ = CxxConfig()
                 config_.set_valid_places(elem_)
                 config_.set_threads(thread_)
                 if target.upper() == "NNADAPTER":
                     self.nnadapter_config_set(self, config_)
                 configs_.append(config_)
     return configs_

Ejemplo n.º 6

 def sample_predictor_configs(self):
     config = CxxConfig()
     config.set_valid_places({
         Place(TargetType.OpenCL, PrecisionType.FP16,
               DataLayoutType.ImageDefault),
         Place(TargetType.OpenCL, PrecisionType.FP16,
               DataLayoutType.ImageFolder),
         Place(TargetType.OpenCL, PrecisionType.FP32, DataLayoutType.NCHW),
         Place(TargetType.OpenCL, PrecisionType.Any,
               DataLayoutType.ImageDefault),
         Place(TargetType.OpenCL, PrecisionType.Any,
               DataLayoutType.ImageFolder),
         Place(TargetType.OpenCL, PrecisionType.Any, DataLayoutType.NCHW),
         Place(TargetType.X86, PrecisionType.FP32),
         Place(TargetType.ARM, PrecisionType.FP32),
         Place(TargetType.Host, PrecisionType.FP32)
     })
     yield config, ['box_coder'], (1e-5, 1e-5)

Ejemplo n.º 7

 def sample_predictor_configs(self):
     config = CxxConfig()
     return self.get_predictor_configs(), [
         "fusion_" + self.ops[0].type + "_activation"
     ], (1e-5, 1e-5)

Ejemplo n.º 8

 def sample_predictor_configs(self):
     config = CxxConfig()
     return self.get_predictor_configs(), ['fc'], (1e-4, 1e-4)

Ejemplo n.º 9

Archivo: test_identity_dropout_eleminate_pass.py Proyecto: zzjjay/Paddle-Lite

 def sample_predictor_configs(self):
     config = CxxConfig()
     return self.get_predictor_configs(), ['hard_swish'], (1e-5, 1e-5)

Ejemplo n.º 10

Archivo: test_remove_tf_redundant_ops_pass.py Proyecto: zzjjay/Paddle-Lite

 def sample_predictor_configs(self):
     config = CxxConfig()
     return self.get_predictor_configs(), ['softmax'], (1e-5, 1e-5)

Ejemplo n.º 11

 def sample_predictor_configs(self):
     config = CxxConfig()
     return self.get_predictor_configs(), ['elementwise_mul'], (1e-5, 1e-5)

Ejemplo n.º 12

 def sample_predictor_configs(self):
     config = CxxConfig()
     config.set_valid_places(
         {Place(TargetType.ARM, PrecisionType.FP32, DataLayoutType.NCHW)})
     config.set_threads(1)
     yield config, ["shuffle_channel"], (1e-5, 1e-5)

Ejemplo n.º 13

Archivo: test_conv2d_op.py Proyecto: liyupeng/Paddle-Lite

 def sample_predictor_configs(self):
     config = CxxConfig()
     return self.get_predictor_configs(), ["conv2d"], (1e-5, 1e-5)

Ejemplo n.º 14

 def sample_predictor_configs(self):
     config = CxxConfig()
     config.set_valid_places(
         {Place(TargetType.ARM, PrecisionType.Any, DataLayoutType.NCHW)})
     config.set_threads(1)
     yield config, ["lookup_table_dequant"], (1e-5, 1e-5)

Ejemplo n.º 15

Archivo: test_fill_constant_batch_size_like_op.py Proyecto: zhangjun/Paddle-Lite

 def sample_predictor_configs(self):
     config = CxxConfig()
     config.set_valid_places(
         {Place(TargetType.Host, PrecisionType.FP32, DataLayoutType.NCHW)})
     yield config, ["fill_constant_batch_size_like"], (1e-5, 1e-5)

Ejemplo n.º 16

 def sample_predictor_configs(self):
     config = CxxConfig()
     config.set_valid_places(
         {Place(TargetType.Host, PrecisionType.Any, DataLayoutType.NCHW)})
     yield config, ["index_select"], (1e-5, 1e-5)

Ejemplo n.º 17

Archivo: test_distribute_fpn_proposals_op.py Proyecto: zzjjay/Paddle-Lite

 def sample_predictor_configs(self):
     config = CxxConfig()
     config.set_valid_places(
         {Place(TargetType.Host, PrecisionType.FP32, DataLayoutType.NCHW)})
     yield config, ["distribute_fpn_proposals"], (1e-5, 1e-5)

Ejemplo n.º 18

Archivo: test_one_hot_v2_op.py Proyecto: zzjjay/Paddle-Lite

 def sample_predictor_configs(self):
     config = CxxConfig()
     config.set_valid_places(
         {Place(TargetType.Host, PrecisionType.FP32, DataLayoutType.NCHW)})
     config.set_threads(1)
     yield config, ["one_hot_v2"], (1e-5, 1e-5)

Ejemplo n.º 19

 def sample_predictor_configs(self):
     config = CxxConfig()
     config.set_valid_places(
         {Place(TargetType.Host, PrecisionType.FP32, DataLayoutType.NCHW)})
     yield config, ["gather_nd"], (1e-5, 1e-5)

Ejemplo n.º 20

 def sample_predictor_configs(self):
     config = CxxConfig()
     config.set_valid_places(
         {Place(TargetType.ARM, PrecisionType.FP32, DataLayoutType.NCHW)})
     yield config, ['softmax'], (1e-5, 1e-5)

Ejemplo n.º 21

 def sample_predictor_configs(self):
     config = CxxConfig()
     config.set_valid_places(
         {Place(TargetType.X86, PrecisionType.FP32, DataLayoutType.NCHW)})
     yield config, ["lookup_table"], (1e-5, 1e-5)

Ejemplo n.º 22

 def sample_predictor_configs(self):
     config = CxxConfig()
     config.set_valid_places(
         {Place(TargetType.ARM, PrecisionType.FP32, DataLayoutType.NCHW)})
     yield config, ["hard_swish", "assign"], (1e-5, 1e-5)

Ejemplo n.º 23

Archivo: test_conv_bn_fuse_pass.py Proyecto: zzjjay/Paddle-Lite

 def sample_predictor_configs(self):
     config = CxxConfig()
     return self.get_predictor_configs(), [self.ops[0].type], (1e-4, 1e-4)

Ejemplo n.º 24

 def sample_predictor_configs(self):
     config = CxxConfig()
     config.set_valid_places(
         {Place(TargetType.Host, PrecisionType.FP32, DataLayoutType.NCHW)})
     yield config, ["flatten_contiguous_range"], (1e-5, 1e-5)