Home
last modified time | relevance | path

Searched defs:include_top (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/st/rewrite/
Dtest_api.py30 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/st/python/import_ms_and_mslite/
Dtest_predict_backend_lite_lenet.py42 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
/third_party/mindspore/mindspore-src/source/tests/st/export_and_load/
Dtest_save_load_checkpoint.py29 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
Dtest_ckpt_d2h_async.py48 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
Dtest_ckpt_bf16.py29 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/train/
Ddataset.py73 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/rewrite/
Dtest_lenet.py24 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
/third_party/mindspore/mindspore-src/source/tests/st/summary/
Dsummary_net.py56 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
Dtest_summary_ops.py35 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
Dtest_summary_collector.py61 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/examples/export_models/models/
Dmini_alexnet.py34 def __init__(self, num_classes=10, channel=1, phase='train', include_top=True): argument
/third_party/mindspore/mindspore-src/source/tests/st/pi_jit/one_stage/
Dtest_network.py35 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
/third_party/mindspore/mindspore-src/source/tests/st/pi_jit/network/
Dtest_lenet.py21 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
Dtest_lenet_inline.py21 def __init__(self, num_class=10, num_channel=1, include_top=True): argument