[Apache TVM Discuss] [Questions] Prelu op can not support broadcast?

2020-09-11 Thread Blacklong via Apache TVM Discuss
Hi,all,I have some problem about ONNX Prelu.I convert my pytorch model to onnx ,but got error: In `main`: #[version = "0.0.5"] fn (%inputL: Tensor[(1, 3, 512, 960), float32], %feature_extraction.firstconv.conv1.0.0.weight: Tensor[(48, 3, 1, 1), float32], %feature_extraction.first

[Apache TVM Discuss] [Questions] TVM opencl context how to choose device type as accelerator

2020-09-11 Thread Zhao Wu via Apache TVM Discuss
The problem is in ```cpp void Init(const std::string& type_key, const std::string& device_type, const std::string& platform_name = ""); virtual void Init() { Init("opencl", "gpu"); } // gpu is the device type ``` Inside `Init`, we will only try to match the `gpu` device type, if