Modify the code specification of dynamic shape operator
This commit is contained in:
parent
ddcf897386
commit
e749e646cc
|
@ -1,5 +1,5 @@
|
|||
/**
|
||||
* Copyright 2021 Huawei Technologies Co., Ltd
|
||||
* Copyright 2020-2022 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.
|
||||
|
|
|
@ -1,5 +1,5 @@
|
|||
/**
|
||||
* Copyright 2021 Huawei Technologies Co., Ltd
|
||||
* Copyright 2020-2022 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.
|
||||
|
|
|
@ -1,5 +1,5 @@
|
|||
/**
|
||||
* Copyright 2021 Huawei Technologies Co., Ltd
|
||||
* Copyright 2020-2022 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.
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
# coding: utf-8
|
||||
|
||||
# Copyright 2020-2021 Huawei Technologies Co., Ltd
|
||||
# Copyright 2020-2022 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.
|
||||
|
@ -5027,7 +5027,7 @@ class SpaceToDepth(PrimitiveWithInfer):
|
|||
return x_dtype
|
||||
|
||||
|
||||
class DepthToSpace(PrimitiveWithInfer):
|
||||
class DepthToSpace(Primitive):
|
||||
r"""
|
||||
Rearrange blocks of depth data into spatial dimensions.
|
||||
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
# Copyright 2020-2021 Huawei Technologies Co., Ltd
|
||||
# Copyright 2020-2022 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.
|
||||
|
@ -6034,7 +6034,7 @@ class SparseApplyProximalAdagrad(PrimitiveWithCheck):
|
|||
validator.check_tensor_dtype_valid('indices', indices_dtype, valid_dtypes, self.name)
|
||||
|
||||
|
||||
class ApplyAddSign(PrimitiveWithInfer):
|
||||
class ApplyAddSign(Primitive):
|
||||
r"""
|
||||
Updates relevant entries according to the AddSign algorithm.
|
||||
|
||||
|
|
Loading…
Reference in New Issue