Searched refs:dense_out_channel (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_pack.py | 76 def __init__(self, dense_in_channel, dense_out_channel, axis=0, shape=None, strategy=None): argument
78 weight_np = np.full((dense_out_channel, dense_in_channel), 0.01, dtype=np.float32)
79 bias_np = np.full((dense_out_channel), 0.01, dtype=np.float32)
83 out_channels=dense_out_channel,
103 def __init__(self, dense_in_channel, dense_out_channel, axis=0, shape=None, strategy=None): argument
105 weight_np = np.full((dense_out_channel, dense_in_channel), 0.01, dtype=np.float32)
106 bias_np = np.full((dense_out_channel), 0.01, dtype=np.float32)
110 out_channels=dense_out_channel,
261 net = PackConstantNet1(dense_in_channel=64, dense_out_channel=4, axis=0, shape=(8, 8),
268 net = PackConstantNet2(dense_in_channel=64, dense_out_channel=4, axis=0, shape=(8, 8),
[all …]
|
| D | test_virtual_output.py | 101 def __init__(self, dense_in_channel=2048, dense_out_channel=250): argument
103 weight_np = np.full((dense_out_channel, dense_in_channel), 0.01, dtype=np.float32)
104 bias_np = np.full((dense_out_channel,), 0.01, dtype=np.float32)
107 out_channels=dense_out_channel,
|
| D | test_reshape.py | 607 def __init__(self, dense_in_channel, dense_out_channel, shape, strategy=None): argument
611 out_channels=dense_out_channel,
636 …net = ParallelReshapeNet(dense_in_channel=2048, dense_out_channel=1000, shape=(128, 1000), strateg…
|