| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mkldnn_fusion.py | 65 def __init__(self, in_channels, out_channels, bias, **kwargs): argument
104 def __init__(self, unary_fn, in_channels, out_channels, bias, **kwargs): argument
133 def __init__(self, m, in_channels, out_channels, bias, **kwargs): argument
196 def __init__(self, unary_fn, in_channels, out_channels, bias, **kwargs): argument
230 … def __init__(self, unary_fn, dim, in_channels, out_channels, dilation, groups, bias, **kwargs): argument
266 … def __init__(self, binary_fn, dim, in_channels, out_channels, dilation, groups, bias, **kwargs): argument
314 def __init__(self, binary_fn, in_channels, out_channels, bias, **kwargs): argument
347 def __init__(self, unary_fn, dim, in_channels, out_channels, kernel_size, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/qat/modules/ |
| H A D | conv_fused.py | 47 out_channels, argument
463 out_channels, argument
534 out_channels, argument
596 out_channels, argument
660 out_channels, argument
731 out_channels, argument
793 out_channels, argument
857 out_channels, argument
927 out_channels, argument
991 out_channels, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
| H A D | conv.py | 44 out_channels, argument
61 out_channels, argument
530 out_channels, argument
659 out_channels, argument
752 out_channels, argument
922 out_channels, argument
1045 out_channels, argument
1170 out_channels, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/ |
| H A D | conv.py | 142 out_channels, argument
226 out_channels, argument
316 out_channels, argument
398 out_channels, argument
480 out_channels, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/reference/modules/ |
| H A D | conv.py | 121 out_channels, argument
184 out_channels, argument
357 out_channels, argument
438 out_channels, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/quantized/modules/ |
| H A D | conv_relu.py | 36 out_channels, argument
117 out_channels, argument
198 out_channels, argument
|
| H A D | conv_add.py | 27 out_channels, argument
95 out_channels, argument
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_inductor_freezing.py | 74 def __init__(self, in_channels, out_channels, bias=False, **kwargs): argument
84 def __init__(self, in_channels, out_channels, bias=False, **kwargs): argument
98 out_channels, argument
130 def __init__(self, in_channels, out_channels, bias=False, **kwargs): argument
146 out_channels, argument
|
| H A D | test_efficient_conv_bn_eval.py | 38 out_channels, argument
62 out_channels, argument
|
| H A D | test_binary_folding.py | 41 def __init__(self, in_channels, out_channels, device, **kwargs): argument
165 def __init__(self, in_channels, out_channels, device, **kwargs): argument
|
| /aosp_15_r20/external/libaom/av1/encoder/arm/ |
| H A D | cnn_neon.c | 781 const int out_channels, float **output, int out_stride, int start_idx, in av1_cnn_convolve_no_maxpool_padding_valid_2x2_large_neon()
881 const int out_channels, float **output, int out_stride, int start_idx, in av1_cnn_convolve_no_maxpool_padding_valid_2x2_neon()
958 const int out_channels, float **output, int out_stride, int start_idx, in av1_cnn_convolve_no_maxpool_padding_valid_5x5_neon()
|
| /aosp_15_r20/external/libopus/dnn/ |
| H A D | nnet_arch.h | 159 static void conv2d_float(float *out, const float *weights, int in_channels, int out_channels, int k… in conv2d_float()
185 static void conv2d_3x3_float(float *out, const float *weights, int in_channels, int out_channels, c… in conv2d_3x3_float()
|
| H A D | nndsp.c | 84 int out_channels, in scale_kernel()
143 int out_channels, in adaconv_process_frame()
|
| /aosp_15_r20/external/webrtc/modules/audio_coding/codecs/opus/test/ |
| H A D | lapped_transform_unittest.cc | 30 size_t out_channels, in ProcessAudioBlock()
52 size_t out_channels, in ProcessAudioBlock()
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Convolution.cpp | 94 const int64_t out_channels = transposed ? sizes.at(1) : sizes.at(0); in prepack_biases() local
441 int32_t out_channels = weight_sizes.at(0); in add_conv1d_node() local
|
| H A D | Permute.cpp | 72 int32_t out_channels = dim_at<kChannel4D>(t_out->sizes()); in add_permute_node() local
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_freezing.py | 2076 def __init__(self, in_channels, out_channels, **kwargs): argument
2129 def __init__(self, in_channels, out_channels, **kwargs): argument
2169 def __init__(self, in_channels, out_channels, **kwargs): argument
2204 def __init__(self, in_channels, out_channels, tensor=None, **kwargs): argument
2320 def __init__(self, in_channels, out_channels, **kwargs): argument
2986 def __init__(self, in_channels, out_channels, **kwargs): argument
3041 def __init__(self, in_channels, out_channels, **kwargs): argument
|
| H A D | test_models.py | 194 def __init__(self, in_channels, out_channels, kernel_size, stride): argument
236 self, in_channels, out_channels, kernel_size, stride, upsample=None argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | quantization_ops_test.cc | 128 Output AddConv2D(Scope scope, Input input, int in_channels, int out_channels, in AddConv2D()
481 std::array<int, 2> out_channels = {16, 32}; in TEST_P() local
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_convolution.py | 3408 out_channels, argument
3614 def helper(n, c, h, w, out_channels, kernel_size, groups): argument
3668 def helper(n, c, d, h, w, out_channels, kernel_size, groups): argument
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_qat.py | 72 in_channels, out_channels, kernel_size, stride, argument
234 in_channels, out_channels, kernel_size, stride=1, argument
|
| /aosp_15_r20/external/pytorch/test/ao/sparsity/ |
| H A D | test_kernels.py | 254 def __init__(self, in_channels, out_channels): argument
|
| /aosp_15_r20/external/libopus/dnn/torch/osce/utils/layers/ |
| H A D | limited_adaptive_conv1d.py | 44 out_channels, argument
|
| /aosp_15_r20/external/pytorch/test/xpu/ |
| H A D | test_conv.py | 999 def helper(n, c, h, w, out_channels, kernel_size, groups): argument
1048 def helper(n, c, d, h, w, out_channels, kernel_size, groups): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | flatten_atrous.cc | 72 const int32_t out_channels = filter.dim_size(3); in FlattenAtrousConv() local
|