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