| /third_party/mindspore/tests/st/quantization/lenet_quant/ |
| D | lenet_fusion.py | 33 def __init__(self, num_class=10, channel=1): argument
|
| /third_party/mindspore/tests/st/gnn/ |
| D | gat.py | 65 num_class, argument
|
| /third_party/mindspore/tests/st/networks/models/deeplabv3/src/ |
| D | miou_precision.py | 34 def __init__(self, num_class=21): argument
|
| /third_party/mindspore/tests/st/fl/mobile/src/ |
| D | model.py | 47 def __init__(self, num_class=10, channel=3): argument
|
| /third_party/mindspore/tests/perf_test/ |
| D | test_lenet.py | 36 num_class = 10 variable
|
| D | lenet.py | 53 def __init__(self, num_class=10, channel=1): argument
|
| /third_party/mindspore/tests/st/summary/ |
| D | test_summary_ops.py | 37 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
|
| D | test_summary_collector.py | 51 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
|
| /third_party/mindspore/tests/st/ps/cmp_sparse_embedding/ |
| D | test_cmp_sparse_embedding.py | 42 def __init__(self, num_class=10): argument
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | model.py | 48 def __init__(self, num_class=10, channel=3): argument
|
| /third_party/mindspore/tests/st/fl/cross_silo_lenet/src/ |
| D | model.py | 48 def __init__(self, num_class=10, channel=3): argument
|
| /third_party/mindspore/tests/st/ps/multi_full_ps/ |
| D | test_multi_full_ps.py | 67 def __init__(self, num_class=10, channel=3): argument
|
| /third_party/mindspore/tests/st/dump/ |
| D | test_data_dump.py | 166 def __init__(self, kernel, bias, in_channel, num_class): argument
|
| /third_party/mindspore/tests/st/probability/toolbox/ |
| D | test_uncertainty.py | 51 def __init__(self, num_class=10, channel=1): argument
|
| /third_party/mindspore/tests/st/armour/ |
| D | test_pynative_mindarmour.py | 60 def __init__(self, num_class=10): argument
|
| /third_party/mindspore/tests/st/probability/bnn_layers/ |
| D | test_bnn_layer.py | 63 def __init__(self, num_class=10): argument
|
| /third_party/mindspore/tests/st/probability/transforms/ |
| D | test_transform_bnn_model.py | 63 def __init__(self, num_class=10): argument
|
| D | test_transform_bnn_layer.py | 64 def __init__(self, num_class=10): argument
|
| /third_party/mindspore/tests/st/fusion/ |
| D | test_conv_bn1_fusion.py | 27 num_class = 10 variable
|
| /third_party/mindspore/tests/st/ps/full_ps/ |
| D | test_full_ps_lenet.py | 61 def __init__(self, num_class=10, channel=1): argument
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_lenet.py | 66 def __init__(self, num_class=10): argument
|
| /third_party/mindspore/tests/st/nontask_sink/ |
| D | test_lenet.py | 65 def __init__(self, num_class=10): argument
|
| /third_party/mindspore/tests/st/profiler/ |
| D | test_profiler.py | 59 def __init__(self, num_class=10, channel=1): argument
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gpu_lenet.py | 62 def __init__(self, num_class=10, channel=1): argument
|
| /third_party/mindspore/tests/st/pynative/ms_function/ |
| D | test_pynative_lenet_ms_function.py | 103 def __init__(self, num_class=10): argument
|