|
|
|
@ -13,6 +13,8 @@
|
|
|
|
|
# limitations under the License.
|
|
|
|
|
# ============================================================================
|
|
|
|
|
"""thor_layer"""
|
|
|
|
|
import numpy as np
|
|
|
|
|
|
|
|
|
|
import mindspore as ms
|
|
|
|
|
import mindspore.common.dtype as mstype
|
|
|
|
|
from mindspore._checkparam import check_bool, twice, check_int_positive
|
|
|
|
@ -23,7 +25,6 @@ from mindspore.common.tensor import Tensor
|
|
|
|
|
from mindspore.nn.cell import Cell
|
|
|
|
|
from mindspore.nn.layer.activation import get_activation
|
|
|
|
|
from mindspore.ops import operations as P
|
|
|
|
|
import numpy as np
|
|
|
|
|
C0 = 16
|
|
|
|
|
|
|
|
|
|
def caculate_device_shape(matrix_dim, channel, is_A):
|
|
|
|
|