|
|
@@ -111,8 +111,7 @@ class AiCoreOpTask { |
|
|
|
Status RegisterKernelHandle(const OpDesc &op_desc); |
|
|
|
Status InitWithKernelDef(const OpDesc &op_desc, const domi::TaskDef &task_def); |
|
|
|
Status InitWithKernelDefWithHandle(const OpDesc &node, const domi::TaskDef &task_def); |
|
|
|
Status UpdateShapeToOutputDesc(TaskContext &context, const GeShape &shape, |
|
|
|
const int output_index) const; |
|
|
|
Status UpdateShapeToOutputDesc(TaskContext &context, const GeShape &shape, const int output_index) const; |
|
|
|
|
|
|
|
std::string stub_name_; |
|
|
|
void *stub_func_ = nullptr; |
|
|
|