test=huawei_ascend_npu

PaddlePaddle · Oct 19, 2021 · 61bde22 · 61bde22
1 parent e30558c
commit 61bde22
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/lite/kernels/nnadapter/converter/expand_v2.cc b/lite/kernels/nnadapter/converter/expand_v2.cc
@@ -49,8 +49,8 @@ int ConvertExpandV2(Converter* converter, OpInfo* op, Scope* scope) {
   std::vector<NNAdapterOperand*> input_operands = {input_operand,
                                                    shape_operand};
   std::vector<NNAdapterOperand*> output_operands = {output_operand};
-  converter->AddOperation(NNADAPTER_EXPAND, &input_operands, &output_operands);
-
+  converter->AddOperation(
+      NNADAPTER_EXPAND, {input_operand, shape_operand}, {output_operand});
   return NO_ERROR;
 }
 

diff --git a/lite/kernels/nnadapter/converter/range.cc b/lite/kernels/nnadapter/converter/range.cc
@@ -58,7 +58,9 @@ int ConvertRange(Converter* converter, OpInfo* op, Scope* scope) {
   std::vector<NNAdapterOperand*> input_operands = {
       start_operand, end_operand, step_operand};
   std::vector<NNAdapterOperand*> output_operands = {output_operand};
-  converter->AddOperation(NNADAPTER_RANGE, &input_operands, &output_operands);
+  converter->AddOperation(NNADAPTER_EXPAND,
+                          {start_operand, end_operand, step_operand},
+                          {output_operand});
   return NO_ERROR;
 }