Searched defs:num_channel (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/lite/examples/export_models/models/ |
| D | NetworkInNetwork.py | 26 def __init__(self, num_classes=10, num_channel=3): 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/networks/ |
| D | test_gradient_accumulation.py | 53 def __init__(self, num_class=10, num_channel=1): argument
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | normalization.py | 243 def _channel_check(channel, num_channel, prim_name=None): argument
|