Searched refs:tensor_engine (Results 1 – 4 of 4) sorted by relevance
1 tensor_engine = None variable19 global tensor_engine23 tensor_engine = tf_engine.TensorFlowEngine()27 tensor_engine = pt_engine.TorchTensorEngine()31 tensor_engine = topi_engine.TopiEngine()35 tensor_engine = relay_engine.RelayEngine()39 tensor_engine = nnc_engine.NncEngine()42 tensor_engine.mode = mode46 if tensor_engine is None:48 return tensor_engine
10 from . import tensor_engine30 self.engine = tensor_engine.get_engine()152 engine = tensor_engine.get_engine()
1 from . import benchmark, tensor_engine66 return tensor_engine.is_supported(self.instance_norm)
19 tensor_engine,221 tensor_engine.set_engine_mode(args.engine)