Python CxxConfig.set_valid_places示例

示例#1

 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)

示例#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)

示例#3

 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)

示例#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_

示例#5

文件： auto_scan_base.py 项目： 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_

示例#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)

示例#7

 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)

示例#8

 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)

示例#9

 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)

示例#10

文件： test_one_hot_v2_op.py 项目： 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)

示例#11

 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)

示例#12

文件： test_fill_constant_batch_size_like_op.py 项目： 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)

示例#13

 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)

示例#14

文件： test_distribute_fpn_proposals_op.py 项目： 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)

示例#15

 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)

示例#16

 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)

示例#17

 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)