Home
last modified time | relevance | path

Searched defs:layer_num (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/st/pynative/network/
Dtest_pynative_resnet50_ascend.py218 def __init__(self, block, layer_num, in_channels, out_channels, stride): argument
235 def __init__(self, block, layer_num, in_channels, out_channels, stride): argument
254 def __init__(self, block, layer_num, in_channels, out_channels, stride): argument
277 def __init__(self, block, layer_num, in_channels, out_channels, stride): argument
294 def __init__(self, block, layer_num, num_classes=100): argument
/third_party/mindspore/mindspore-src/source/tests/st/export_and_load/
Dtest_resnet_air.py207 def __init__(self, block, layer_num, in_channels, out_channels, stride): argument
224 def __init__(self, block, layer_num, in_channels, out_channels, stride): argument
243 def __init__(self, block, layer_num, in_channels, out_channels, stride): argument
266 def __init__(self, block, layer_num, in_channels, out_channels, stride): argument
283 def __init__(self, block, layer_num, num_classes=100): argument
/third_party/mindspore/mindspore-src/source/tests/st/networks/
Dtest_gpu_resnet.py211 def __init__(self, block, layer_num, in_channels, out_channels, stride): argument
228 def __init__(self, block, layer_num, in_channels, out_channels, stride): argument
247 def __init__(self, block, layer_num, in_channels, out_channels, stride): argument
270 def __init__(self, block, layer_num, in_channels, out_channels, stride): argument
287 def __init__(self, block, layer_num, num_classes=100): argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/model/
Dres18_example.py124 def MakeLayer(self, block, layer_num, in_channels, out_channels, stride): argument
193 def MakeLayer(self, block, layer_num, in_channels, out_channels, stride): argument
Dresnet.py198 def _make_layer(self, block, layer_num, in_channel, out_channel, stride): argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Dcle_pattern.h30 size_t layer_num = 0; member
Dgptq_quantizer.cc324 size_t layer_num = dynamic_session_->GetKernelNum(); in RunKernel() local
/third_party/mindspore/mindspore-src/source/tests/perf_test/
Dresnet_example.py121 def MakeLayer(self, block, layer_num, in_channels, out_channels, stride): argument
/third_party/mindspore/mindspore-src/source/tests/st/compile_cache/
Drun_resnet_infer.py125 def MakeLayer(self, block, layer_num, in_channels, out_channels, stride): argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/rewrite/
Dtest_subtree_resnet.py10 def make_layer(block, layer_num, in_channel, out_channel, stride, use_se=False, se_block=False): argument
Dtest_cellcontainer.py94 def _make_layer(self, block, layer_num, in_channel, out_channel, stride): argument
/third_party/mindspore/mindspore-src/source/tests/st/networks/models/resnet50/src/
Dresnet.py198 def _make_layer(self, block, layer_num, in_channel, out_channel, stride): argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/
Dtest_auto_parallel_resnet_sharding_propagation2.py181 def _make_layer(self, block, layer_num, in_channel, out_channel, stride): argument
Dtest_auto_parallel_resnet_sharding_propagation.py182 def _make_layer(self, block, layer_num, in_channel, out_channel, stride): argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/st/python/import_ms_and_mslite/
Dresnet.py432 … def _make_layer(block, layer_num, in_channel, out_channel, stride, use_se=False, se_block=False): argument
/third_party/mindspore/mindspore-src/source/tests/st/hypercomplex/
Dresnet.py455 …def _make_layer(self, block, layer_num, in_channel, out_channel, stride, use_se=False, se_block=Fa… argument
/third_party/mindspore/mindspore-src/source/tests/ut/python/debug/
Dresnet.py455 …def _make_layer(self, block, layer_num, in_channel, out_channel, stride, use_se=False, se_block=Fa… argument
/third_party/mindspore/mindspore-src/source/tests/st/auto_parallel/
Dcell_shard.py208 def _make_layer(self, block, layer_num, in_channel, out_channel, stride): argument
Dresnet50_expand_loss.py209 def _make_layer(self, block, layer_num, in_channel, out_channel, stride): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/
Dstage_compute.cc478 … size_t head_num, size_t layer_num, size_t expansion_ratio, size_t dp, size_t mp, in StageComputing()