Python FallbackConfigEntity.template_key示例

示例#1

文件： test_topi_conv2d_winograd.py 项目： zhangquan920/tasn

 def _query_inside(self, target, workload):
     key = (target, workload)
     if key in self.memory:
         return self.memory[key]
     cfg = FallbackConfigEntity()
     cfg.template_key = 'winograd'
     self.memory[key] = cfg
     return cfg

示例#2

文件： common.py 项目： LANHUIYING/tvm

 def _query_inside(self, target, workload):
     key = (target, workload)
     if key in self.memory:
         return self.memory[key]
     cfg = FallbackConfigEntity()
     cfg.template_key = 'int8'
     self.memory[key] = cfg
     return cfg

示例#3

 def _query_inside(self, target, workload):
     key = (target, workload)
     if key in self.memory:
         return self.memory[key]
     cfg = FallbackConfigEntity()
     cfg.template_key = 'int8'
     self.memory[key] = cfg
     cfg.is_fallback = False
     return cfg