forked from huawei/mindspore2022
!8797 move declaration to correct headers
From: @zhaozhenlong Reviewed-by: @zhang_xue_tong,@hangangqiang Signed-off-by: @zhang_xue_tong
This commit is contained in:
commit
1145fd1ca9
|
|
@ -20,7 +20,10 @@
|
|||
|
||||
namespace mindspore {
|
||||
namespace lite {
|
||||
|
||||
ArithmeticParameter *PopulateArithmeticCommonPara(const mindspore::lite::PrimitiveC *primitive);
|
||||
OpParameter *PopulateArithmetic(const mindspore::lite::PrimitiveC *primitive);
|
||||
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_SRC_OPS_POPULATE_ARITHMETIC_POPULATE_H_
|
||||
|
|
|
|||
|
|
@ -21,6 +21,7 @@
|
|||
|
||||
namespace mindspore {
|
||||
namespace lite {
|
||||
|
||||
class PopulateRegistry {
|
||||
public:
|
||||
static PopulateRegistry *GetInstance() {
|
||||
|
|
@ -50,8 +51,7 @@ class Registry {
|
|||
}
|
||||
~Registry() = default;
|
||||
};
|
||||
OpParameter *PopulateArithmetic(const mindspore::lite::PrimitiveC *primitive);
|
||||
OpParameter *PopulateStridedSliceParameter(const mindspore::lite::PrimitiveC *primitive);
|
||||
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
#endif
|
||||
|
|
|
|||
|
|
@ -14,6 +14,7 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/ops/populate/strided_slice_populate.h"
|
||||
#include "src/ops/strided_slice.h"
|
||||
#include "src/ops/primitive_c.h"
|
||||
#include "src/ops/populate/populate_register.h"
|
||||
|
|
@ -46,5 +47,6 @@ OpParameter *PopulateStridedSliceParameter(const mindspore::lite::PrimitiveC *pr
|
|||
}
|
||||
|
||||
Registry StridedSliceParameterRegistry(schema::PrimitiveType_StridedSlice, PopulateStridedSliceParameter);
|
||||
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
|
|
|
|||
|
|
@ -0,0 +1,25 @@
|
|||
/**
|
||||
* Copyright 2019-2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "src/ops/arithmetic.h"
|
||||
|
||||
namespace mindspore {
|
||||
namespace lite {
|
||||
|
||||
OpParameter *PopulateStridedSliceParameter(const mindspore::lite::PrimitiveC *primitive);
|
||||
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
|
|
@ -15,6 +15,7 @@
|
|||
*/
|
||||
|
||||
#include "src/ops/strided_slice.h"
|
||||
#include "src/ops/populate/strided_slice_populate.h"
|
||||
#include <algorithm>
|
||||
|
||||
#ifndef PRIMITIVE_WRITEABLE
|
||||
|
|
|
|||
|
|
@ -20,7 +20,7 @@
|
|||
#include "src/kernel_registry.h"
|
||||
#include "include/errorcode.h"
|
||||
#include "src/runtime/runtime_api.h"
|
||||
#include "src/ops/populate/populate_register.h"
|
||||
#include "src/ops/populate/strided_slice_populate.h"
|
||||
|
||||
using mindspore::kernel::KERNEL_ARCH::kCPU;
|
||||
using mindspore::lite::KernelRegistrar;
|
||||
|
|
|
|||
|
|
@ -21,7 +21,7 @@
|
|||
#include "schema/model_generated.h"
|
||||
#include "src/kernel_registry.h"
|
||||
#include "src/runtime/runtime_api.h"
|
||||
#include "src/ops/populate/populate_register.h"
|
||||
#include "src/ops/populate/arithmetic_populate.h"
|
||||
#include "include/errorcode.h"
|
||||
|
||||
using mindspore::kernel::KERNEL_ARCH::kCPU;
|
||||
|
|
|
|||
|
|
@ -20,7 +20,7 @@
|
|||
#include "src/kernel_registry.h"
|
||||
#include "src/runtime/kernel/arm/int8/add_int8.h"
|
||||
#include "src/runtime/runtime_api.h"
|
||||
#include "src/ops/populate/populate_register.h"
|
||||
#include "src/ops/populate/arithmetic_populate.h"
|
||||
|
||||
using mindspore::kernel::KERNEL_ARCH::kCPU;
|
||||
using mindspore::lite::KernelRegistrar;
|
||||
|
|
|
|||
Loading…
Reference in New Issue