Home
last modified time | relevance | path

Searched defs:in_channels (Results 1 – 25 of 58) sorted by relevance

123

/aosp_15_r20/external/pytorch/test/
H A 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
/aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/qat/modules/
H A Dconv_fused.py46 in_channels, argument
462 in_channels, argument
533 in_channels, argument
595 in_channels, argument
659 in_channels, argument
730 in_channels, argument
792 in_channels, argument
856 in_channels, argument
926 in_channels, argument
990 in_channels, argument
/aosp_15_r20/external/pytorch/test/fx/
H A Dtest_fx_param_shape_control_flow.py26 def __init__(self, in_channels): argument
35 def __init__(self, in_channels): argument
68 def __init__(self, in_channels): argument
77 def __init__(self, in_channels): argument
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/
H A Dconv.py43 in_channels, argument
60 in_channels, argument
529 in_channels, argument
658 in_channels, argument
751 in_channels, argument
921 in_channels, argument
1044 in_channels, argument
1169 in_channels, argument
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/
H A Dconv.py141 in_channels, argument
225 in_channels, argument
315 in_channels, argument
397 in_channels, argument
479 in_channels, argument
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/reference/modules/
H A Dconv.py120 in_channels, argument
183 in_channels, argument
356 in_channels, argument
437 in_channels, argument
/aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/quantized/modules/
H A Dconv_relu.py35 in_channels, argument
116 in_channels, argument
197 in_channels, argument
H A Dconv_add.py26 in_channels, argument
94 in_channels, argument
/aosp_15_r20/external/pytorch/test/inductor/
H A 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
97 in_channels, argument
130 def __init__(self, in_channels, out_channels, bias=False, **kwargs): argument
145 in_channels, argument
H A Dtest_efficient_conv_bn_eval.py37 in_channels, argument
61 in_channels, argument
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_spatial_convolutions_test.cc668 const int in_channels = 10; in TEST() local
729 const int in_channels = 10; in TEST() local
791 const int in_channels = 10; in TEST() local
843 const int in_channels = 10; in TEST() local
897 const int in_channels = 10; in TEST() local
963 const int in_channels = 10; in TEST() local
1029 const int in_channels = 10; in TEST() local
1084 const int in_channels = 10; in TEST() local
1140 const int in_channels = 10; in TEST() local
1210 const int in_channels = 10; in TEST() local
/aosp_15_r20/external/libaom/av1/encoder/arm/
H A Dcnn_neon.c780 const int filter_width, const int filter_height, const int in_channels, in av1_cnn_convolve_no_maxpool_padding_valid_2x2_large_neon()
880 const int filter_width, const int filter_height, const int in_channels, in av1_cnn_convolve_no_maxpool_padding_valid_2x2_neon()
957 const int filter_width, const int filter_height, const int in_channels, in av1_cnn_convolve_no_maxpool_padding_valid_5x5_neon()
/aosp_15_r20/external/libopus/dnn/
H A 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()
H A Dnndsp.c83 int in_channels, in scale_kernel()
142 int in_channels, in adaconv_process_frame()
/aosp_15_r20/external/webrtc/modules/audio_coding/codecs/opus/test/
H A Dlapped_transform_unittest.cc28 size_t in_channels, in ProcessAudioBlock()
50 size_t in_channels, in ProcessAudioBlock()
/aosp_15_r20/external/libxaac/decoder/
H A Dixheaacd_mps_initfuncs.c1001 WORD32 in_channels = pstr_mps_state->num_input_channels; in ixheaacd_modules_init() local
1086 WORD32 in_channels = pstr_mps_state->num_input_channels; in ixheaacd_aac_mps_create() local
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_freezing.py2076 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 Dtest_models.py194 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 Dquantization_ops_test.cc128 Output AddConv2D(Scope scope, Input input, int in_channels, int out_channels, in AddConv2D()
480 std::array<int, 2> in_channels = {3, 16}; in TEST_P() local
/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_quantize_eager_qat.py72 in_channels, out_channels, kernel_size, stride, argument
234 in_channels, out_channels, kernel_size, stride=1, argument
/aosp_15_r20/external/executorch/examples/models/llama3_2_vision/vision_encoder/
H A Dmodel.py31 in_channels: int = 3 variable in VisionEncoderConfig
/aosp_15_r20/external/pytorch/test/ao/sparsity/
H A Dtest_kernels.py254 def __init__(self, in_channels, out_channels): argument
/aosp_15_r20/external/libopus/dnn/torch/osce/utils/layers/
H A Dlimited_adaptive_conv1d.py43 in_channels, argument
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DPermute.cpp73 int32_t in_channels = dim_at<kChannel4D>(t_in->sizes()); in add_permute_node() local

123