diff --git a/mindspore-lite/src/common/ops/operator_populate/base_operator_populate.cc b/mindspore-lite/src/common/ops/operator_populate/base_operator_populate.cc index 406854a354622783fe1f0bdbe35e04ef28bfa553..c92a4e620b8082b5509c358060c40c0c818204e1 100644 --- a/mindspore-lite/src/common/ops/operator_populate/base_operator_populate.cc +++ b/mindspore-lite/src/common/ops/operator_populate/base_operator_populate.cc @@ -37,7 +37,7 @@ #include "infer/sparse_fill_empty_rows.h" #include "infer/size.h" #include "infer/scatter_nd_update.h" -#include "infer/tensor_scatter_add.h" +#include "mindspore/ops/op_def/auto_generate/gen_ops_name_t.h" #include "infer/ragged_range.h" #include "infer/ops_func_impl/isfinite.h" #include "infer/invert_permutation.h" diff --git a/mindspore-lite/src/common/ops/ops_func_declare.h b/mindspore-lite/src/common/ops/ops_func_declare.h index 2ed4481658d36736bed3cdfc5bee1b8816a1a9bd..d1563e757cdbd445aa4e3052b53fdfe7e6635e15 100644 --- a/mindspore-lite/src/common/ops/ops_func_declare.h +++ b/mindspore-lite/src/common/ops/ops_func_declare.h @@ -170,7 +170,7 @@ #include "infer/dynamic_quant.h" #include "infer/random_normal.h" #include "infer/format_transpose.h" -#include "infer/tensor_scatter_add.h" +#include "mindspore/ops/op_def/auto_generate/gen_ops_name_t.h" #include "infer/decoder_layer.h" #include "infer/encoder_layer.h" #include "infer/scatter_elements.h" diff --git a/mindspore-lite/tools/converter/adapter/acl/mapper/tensor_scatter_mapper.cc b/mindspore-lite/tools/converter/adapter/acl/mapper/tensor_scatter_mapper.cc index e9cf2c1c31558640d8f8a99578a58f3958197e41..ff3c4702f8bf20be7e055e3c47271958ae889c9a 100644 --- a/mindspore-lite/tools/converter/adapter/acl/mapper/tensor_scatter_mapper.cc +++ b/mindspore-lite/tools/converter/adapter/acl/mapper/tensor_scatter_mapper.cc @@ -18,7 +18,8 @@ #include #include "tools/converter/adapter/acl/mapper/primitive_mapper_register.h" #include "src/common/log_util.h" -#include "infer/tensor_scatter_add.h" +#include "mindspore/ops/op_def/auto_generate/gen_lite_ops.h" +#include "mindspore/ops/op_def/auto_generate/gen_ops_name_t.h" #include "ops_utils/op_utils.h" namespace mindspore { diff --git a/mindspore-lite/tools/converter/adapter/acl/mapper/tensor_scatter_mapper.h b/mindspore-lite/tools/converter/adapter/acl/mapper/tensor_scatter_mapper.h index 2f647879e8666ea162ceb5aca34285662ae545cf..6b270c0ffa9bf115560d7aa511ce98f4d1c9dd76 100644 --- a/mindspore-lite/tools/converter/adapter/acl/mapper/tensor_scatter_mapper.h +++ b/mindspore-lite/tools/converter/adapter/acl/mapper/tensor_scatter_mapper.h @@ -18,7 +18,8 @@ #define MINDSPORE_LITE_TOOLS_CONVERTER_ADAPTER_ACL_MAPPER_TENSOR_SCATTER_MAPPER_H_ #include "tools/converter/adapter/acl/mapper/primitive_mapper.h" -#include "infer/tensor_scatter_add.h" +#include "mindspore/ops/op_def/auto_generate/gen_lite_ops.h" +#include "mindspore/ops/op_def/auto_generate/gen_ops_name_t.h" namespace mindspore { namespace lite {