Home
last modified time | relevance | path

Searched defs:out_channels (Results 1 – 25 of 60) sorted by relevance

123

/external/pytorch/test/
Dtest_mkldnn_fusion.py65 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
/external/pytorch/torch/ao/nn/intrinsic/qat/modules/
Dconv_fused.py47 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
/external/pytorch/torch/ao/nn/quantized/dynamic/modules/
Dconv.py142 out_channels, argument
226 out_channels, argument
316 out_channels, argument
398 out_channels, argument
480 out_channels, argument
/external/pytorch/test/inductor/
Dtest_efficient_conv_bn_eval.py38 out_channels, argument
62 out_channels, argument
Dtest_binary_folding.py41 def __init__(self, in_channels, out_channels, device, **kwargs): argument
165 def __init__(self, in_channels, out_channels, device, **kwargs): argument
Dtest_inductor_freezing.py74 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
/external/pytorch/torch/ao/nn/intrinsic/quantized/modules/
Dconv_relu.py36 out_channels, argument
117 out_channels, argument
198 out_channels, argument
Dconv_add.py27 out_channels, argument
95 out_channels, argument
/external/pytorch/torch/ao/nn/quantized/modules/
Dconv.py44 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
/external/pytorch/torch/ao/nn/quantized/reference/modules/
Dconv.py121 out_channels, argument
184 out_channels, argument
357 out_channels, argument
438 out_channels, argument
/external/libopus/dnn/
Dnnet_arch.h159 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()
Dnndsp.c84 int out_channels, in scale_kernel()
143 int out_channels, in adaconv_process_frame()
Dparse_lpcnet_weights.c183 int out_channels, in conv2d_init()
/external/webrtc/modules/audio_coding/codecs/opus/test/
Dlapped_transform_unittest.cc30 size_t out_channels, in ProcessAudioBlock()
52 size_t out_channels, in ProcessAudioBlock()
/external/pytorch/test/ao/sparsity/
Dtest_kernels.py254 def __init__(self, in_channels, out_channels): argument
/external/libopus/dnn/torch/osce/utils/layers/
Dlimited_adaptive_conv1d.py44 out_channels, argument
/external/libaom/av1/encoder/arm/
Dcnn_neon.c781 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()
/external/executorch/backends/vulkan/runtime/graph/ops/impl/
DPermute.cpp72 int32_t out_channels = dim_at<kChannel4D>(t_out->sizes()); in add_permute_node() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dflatten_atrous.cc72 const int32_t out_channels = filter.dim_size(3); in FlattenAtrousConv() local
/external/pytorch/test/dynamo/
Dtest_verify_correctness.py27 def __init__(self, in_channels, out_channels, **kwargs): argument
/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/
Dqlinear.cpp51 int64_t out_channels = static_cast<int64_t>(packW->R); in apply_impl() local
/external/pytorch/aten/src/ATen/native/vulkan/ops/
DPermute.cpp23 uint32_t out_channels = out_size.data[1u]; in permute_4d() local
DTranspose.cpp24 uint32_t out_channels = out_size.data[1u]; in transpose_4d() local
/external/pytorch/test/distributed/fsdp/
Dtest_fsdp_freezing_weights.py105 self, in_channels, out_channels, with_fsdp, freeze_after_wrap_fsdp argument
/external/webrtc/modules/audio_coding/test/
DTestStereo.cc493 int out_channels, in Run()

123