Searched defs:conv2d_weight (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_resizebilinear.py | 28 def __init__(self, conv2d_weight, out_channel, kernel_size, pad_mode, stride, argument
46 def __init__(self, conv2d_weight, out_channel, kernel_size, pad_mode, stride, argument
63 def __init__(self, conv2d_weight, out_channel, kernel_size, pad_mode, stride, argument
|
| D | test_conv2d_transpose.py | 26 def __init__(self, conv2d_weight, out_channel, kernel_size, pad_mode, stride, argument
41 def __init__(self, conv2d_weight, out_channel, kernel_size, pad_mode, stride, argument
|
| D | test_maxpool_avgpool.py | 26 …def __init__(self, conv2d_weight, out_channel, kernel_size, pad_mode, stride, pool_kernel_size, po… argument
41 …def __init__(self, conv2d_weight, out_channel, kernel_size, pad_mode, stride, pool_kernel_size, po… argument
|
| D | test_batchnorm.py | 25 def __init__(self, conv2d_weight, out_channel, kernel_size, pad_mode, stride, argument
|
| D | test_print.py | 25 def __init__(self, conv2d_weight, out_channel, kernel_size, pad_mode, stride, argument
|
| D | test_conv2d.py | 26 … def __init__(self, conv2d_weight, out_channel, kernel_size, pad_mode, stride, dilation=1, group=1, argument
|