| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_apply_adam.py | 27 … def __init__(self, batch_num, input_channels, output_channels, epoch, lr, weight_decay, epsilon): argument
29 self.input_channels = input_channels
37 input_ = Tensor(np.random.randn(self.batch_num, self.input_channels).astype(np.float32))
38 … weight_np = Tensor(np.random.randn(self.output_channels, self.input_channels).astype(np.float32))
46 ms_dense = Dense(in_channels=self.input_channels,
71 …fact = Adam(batch_num=8, input_channels=20, output_channels=5, epoch=5, lr=0.1, weight_decay=0.0, …
|
| /third_party/skia/third_party/skcms/ |
| D | skcms.cc | 582 uint8_t input_channels [ 1]; member
609 a2b-> input_channels = mftTag-> input_channels[0]; in read_mft_common()
617 if (a2b->input_channels < 1 || a2b->input_channels > ARRAY_COUNT(a2b->input_curves)) { in read_mft_common()
621 for (uint32_t i = 0; i < a2b->input_channels; ++i) { in read_mft_common()
635 b2a-> input_channels = mftTag-> input_channels[0]; in read_mft_common()
640 if (b2a->input_channels != ARRAY_COUNT(b2a->input_curves)) { in read_mft_common()
648 for (uint32_t i = 0; i < b2a->input_channels; ++i) { in read_mft_common()
666 uint32_t byte_len_all_input_tables = out->input_channels * byte_len_per_input_table; in init_tables()
670 for (uint32_t axis = 0; axis < out->input_channels; ++axis) { in init_tables()
678 for (uint32_t i = 0; i < out->input_channels; ++i) { in init_tables()
[all …]
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_gather_v2_primitive.py | 90 def fc_with_initialize(input_channels, out_channels): argument
91 return Dense(input_channels, out_channels)
134 input_channels = 256
140 predict = Tensor(np.ones([batch_size_per_device, input_channels]), dtype=ms.float32)
143 network = fc_with_initialize(input_channels, out_channels)
|
| D | test_operator_model_parallel.py | 48 input_channels, argument
58 self.input_channels = input_channels
63 weight_init, [output_channels, input_channels]), name="weight")
171 def fc_with_initialize(input_channels, out_channels): argument
172 weight_shape = (out_channels, input_channels)
177 return DenseWrap(input_channels, out_channels, weight, bias, has_bias=True,
|
| /third_party/mindspore/tests/ut/python/exec/ |
| D | test_dense.py | 27 input_channels, argument
33 self.fc = nn.Dense(input_channels,
|
| /third_party/flutter/skia/third_party/skcms/ |
| D | skcms.cc | 419 uint8_t input_channels [ 1]; member
447 a2b->input_channels = mftTag->input_channels[0]; in read_mft_common()
455 if (a2b->input_channels < 1 || a2b->input_channels > ARRAY_COUNT(a2b->input_curves)) { in read_mft_common()
459 for (uint32_t i = 0; i < a2b->input_channels; ++i) { in read_mft_common()
478 uint32_t byte_len_all_input_tables = a2b->input_channels * byte_len_per_input_table; in init_a2b_tables()
482 for (uint32_t axis = 0; axis < a2b->input_channels; ++axis) { in init_a2b_tables()
490 for (uint32_t i = 0; i < a2b->input_channels; ++i) { in init_a2b_tables()
594 uint8_t input_channels [ 1]; member
618 a2b->input_channels = mABTag->input_channels[0]; in read_tag_mab()
626 if (a2b->input_channels > ARRAY_COUNT(a2b->input_curves)) { in read_tag_mab()
[all …]
|
| /third_party/mindspore/tests/st/fl/mobile/src/ |
| D | model.py | 34 def fc_with_initialize(input_channels, out_channels): argument
38 return nn.Dense(input_channels, out_channels, weight, bias)
|
| /third_party/mindspore/tests/perf_test/ |
| D | lenet.py | 28 def fc_with_initialize(input_channels, out_channels): argument
32 return nn.Dense(input_channels, out_channels, weight, bias)
|
| /third_party/mindspore/mindspore/lite/examples/export_models/models/ |
| D | mini_alexnet.py | 26 def fc_with_initialize(input_channels, out_channels, has_bias=True): argument
27 return nn.Dense(input_channels, out_channels, has_bias=has_bias)
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | model.py | 35 def fc_with_initialize(input_channels, out_channels): argument
39 return nn.Dense(input_channels, out_channels, weight, bias)
|
| /third_party/mindspore/tests/st/fl/cross_silo_lenet/src/ |
| D | model.py | 35 def fc_with_initialize(input_channels, out_channels): argument
39 return nn.Dense(input_channels, out_channels, weight, bias)
|
| /third_party/mindspore/tests/st/ps/multi_full_ps/ |
| D | test_multi_full_ps.py | 54 def fc_with_initialize(input_channels, out_channels): argument
58 return nn.Dense(input_channels, out_channels, weight, bias)
|
| /third_party/mindspore/tests/ut/python/utils/ |
| D | test_export.py | 25 def fc_with_initialize(input_channels, out_channels): argument
28 return nn.Dense(input_channels, out_channels, weight, bias)
|
| /third_party/mindspore/tests/st/probability/transforms/ |
| D | test_transform_bnn_model.py | 38 def fc_with_initialize(input_channels, out_channels): argument
42 return nn.Dense(input_channels, out_channels, weight, bias)
|
| D | test_transform_bnn_layer.py | 39 def fc_with_initialize(input_channels, out_channels): argument
43 return nn.Dense(input_channels, out_channels, weight, bias)
|
| /third_party/mindspore/tests/ut/python/nn/ |
| D | test_dense.py | 93 input_channels, argument
100 self.dense = nn.Dense(input_channels,
|
| /third_party/mindspore/tests/st/probability/bnn_layers/ |
| D | test_bnn_layer.py | 38 def fc_with_initialize(input_channels, out_channels): argument
42 return nn.Dense(input_channels, out_channels, weight, bias)
|
| /third_party/mindspore/tests/st/probability/toolbox/ |
| D | test_uncertainty.py | 38 def fc_with_initialize(input_channels, out_channels): argument
42 return nn.Dense(input_channels, out_channels, weight, bias)
|
| /third_party/mindspore/tests/st/export_and_load/ |
| D | test_train_mindir.py | 39 def fc_with_initialize(input_channels, out_channels): argument
42 return nn.Dense(input_channels, out_channels, weight, bias)
|
| /third_party/mindspore/tests/st/ps/full_ps/ |
| D | test_full_ps_lenet.py | 48 def fc_with_initialize(input_channels, out_channels): argument
52 return nn.Dense(input_channels, out_channels, weight, bias)
|
| /third_party/mindspore/tests/st/armour/ |
| D | test_pynative_mindarmour.py | 42 def fc_with_initialize(input_channels, out_channels): argument
46 return nn.Dense(input_channels, out_channels, weight, bias)
|
| /third_party/skia/include/third_party/skcms/ |
| D | skcms.h | 115 uint32_t input_channels; member
135 uint32_t input_channels; member
|
| /third_party/mindspore/tests/st/nontask_sink/ |
| D | test_lenet.py | 47 def fc_with_initialize(input_channels, out_channels): argument
51 return nn.Dense(input_channels, out_channels, weight, bias)
|
| /third_party/mindspore/tests/st/control/ |
| D | test_if_mindir.py | 39 def fc_with_initialize(input_channels, out_channels): argument
42 return nn.Dense(input_channels, out_channels, weight, bias)
|
| /third_party/mindspore/tests/st/networks/models/ |
| D | resnetv1_5.py | 83 def fc_with_initialize(input_channels, out_channels): argument
84 weight_shape = (out_channels, input_channels)
89 return nn.Dense(input_channels, out_channels, weight, bias)
|