|
|
@@ -172,6 +172,7 @@ class AiCpuBaseTask : public OpTask { |
|
|
|
std::vector<GeTensorDesc> &output_desc, |
|
|
|
std::vector<DataBuffer> &output_buffers, |
|
|
|
rtStream_t stream) override; |
|
|
|
virtual Status LaunchKernel(rtStream_t stream) = 0; |
|
|
|
protected: |
|
|
|
Status UpdateIoAddr(const std::vector<DataBuffer> &inputs, const std::vector<DataBuffer> &outputs); |
|
|
|
Status SetInputConst(); |
|
|
|