Home
last modified time | relevance | path

Searched defs:input_channels (Results 1 – 25 of 104) sorted by relevance

12345

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dfully_connected_test.cc37 const auto input_channels = channels_rng(); variable
56 const auto input_channels = channels_rng(); variable
79 const auto input_channels = channels_rng(); variable
101 const auto input_channels = channels_rng(); variable
125 const auto input_channels = channels_rng(); variable
148 const auto input_channels = channels_rng(); variable
171 const auto input_channels = channels_rng(); variable
196 const auto input_channels = channels_rng(); variable
220 const auto input_channels = channels_rng(); variable
243 const auto input_channels = channels_rng(); in TEST() local
[all …]
H A Dsigned_quantized_fully_connected_test.cc41 const auto input_channels = channels_rng(); variable
66 const auto input_channels = channels_rng(); variable
95 const auto input_channels = channels_rng(); variable
123 const auto input_channels = channels_rng(); variable
153 const auto input_channels = channels_rng(); variable
182 const auto input_channels = channels_rng(); variable
211 const auto input_channels = channels_rng(); variable
242 const auto input_channels = channels_rng(); variable
272 const auto input_channels = channels_rng(); variable
301 const auto input_channels = channels_rng(); in TEST() local
[all …]
H A Dunsigned_quantized_fully_connected_test.cc43 const auto input_channels = channels_rng(); variable
71 const auto input_channels = channels_rng(); variable
103 const auto input_channels = channels_rng(); variable
134 const auto input_channels = channels_rng(); variable
167 const auto input_channels = channels_rng(); variable
199 const auto input_channels = channels_rng(); variable
231 const auto input_channels = channels_rng(); variable
265 const auto input_channels = channels_rng(); variable
298 const auto input_channels = channels_rng(); variable
330 const auto input_channels = channels_rng(); in TEST() local
[all …]
/aosp_15_r20/external/webrtc/audio/utility/
H A Dchannel_mixing_matrix_unittest.cc54 const int input_channels = ChannelLayoutToChannelCount(input_layout); in TEST() local
181 const int input_channels = ChannelLayoutToChannelCount(input_layout); in TEST() local
211 const int input_channels = ChannelLayoutToChannelCount(input_layout); in TEST() local
243 const int input_channels = ChannelLayoutToChannelCount(input_layout); in TEST() local
275 const int input_channels = ChannelLayoutToChannelCount(input_layout); in TEST() local
335 const int input_channels = ChannelLayoutToChannelCount(input_layout); in TEST() local
375 const int input_channels = ChannelLayoutToChannelCount(input_layout); in TEST() local
410 const int input_channels = ChannelLayoutToChannelCount(input_layout); in TEST() local
445 int input_channels; in TEST() member
454 int input_channels = test_case[n].input_channels; in TEST() local
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/
H A Dconv.h184 int64_t input_channels, in Conv1dImpl()
216 int64_t input_channels, in Conv2dImpl()
251 int64_t input_channels, in Conv3dImpl()
343 int64_t input_channels, in ConvTranspose1dImpl()
385 int64_t input_channels, in ConvTranspose2dImpl()
427 int64_t input_channels, in ConvTranspose3dImpl()
/aosp_15_r20/external/XNNPACK/test/
H A Ddeconvolution-nhwc.cc60 for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) { variable
188 for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) { in TEST() local
323 for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) { in TEST() local
461 for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) { in TEST() local
699 for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) { variable
1010 for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) { in TEST() local
1333 for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) { in TEST() local
1665 for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) { in TEST() local
2074 for (size_t input_channels = 1; input_channels <= 16; input_channels *= 4) { variable
2376 for (size_t input_channels = 14; input_channels <= 20; input_channels++) { in TEST() local
[all …]
H A Dpad-microkernel-tester.h35 inline PadMicrokernelTester& input_channels(size_t input_channels) { in input_channels() function
41 inline size_t input_channels() const { in input_channels() function
H A Dspace-to-depth-operator-tester.h72 inline SpaceToDepthOperatorTester& input_channels(size_t input_channels) { in input_channels() function
78 inline size_t input_channels() const { in input_channels() function
H A Dconv-hwc-microkernel-tester.h126 inline ConvHWCMicrokernelTester& input_channels(size_t input_channels) { in input_channels() function
132 inline size_t input_channels() const { in input_channels() function
H A Dspace-to-depth-nhwc.cc67 for (size_t input_channels = 1; input_channels <= 15; input_channels++) { in TEST() local
166 for (size_t input_channels = 1; input_channels <= 15; input_channels++) { in TEST() local
265 for (size_t input_channels = 1; input_channels <= 15; input_channels++) { in TEST() local
H A Dconv-hwc2chw-microkernel-tester.h128 inline ConvHWC2CHWMicrokernelTester& input_channels(size_t input_channels) { in input_channels() argument
134 inline size_t input_channels() const { in input_channels() function
H A Dconvolution-test-helpers.cc281 size_t input_channels, in compute_depthwise_convolution_qs8_reference_results()
346 size_t input_channels, in compute_depthwise_convolution_qs8_reference_results()
398 size_t input_channels, in compute_depthwise_convolution_qu8_reference_results()
464 size_t input_channels, in compute_depthwise_convolution_qu8_reference_results()
H A Dfully-connected-operator-tester.h35 inline FullyConnectedOperatorTester& input_channels(size_t input_channels) { in input_channels() function
41 inline size_t input_channels() const { in input_channels() function
/aosp_15_r20/external/XNNPACK/src/operators/
H A Dfully-connected-nc.c28 size_t input_channels, in create_fully_connected_nc()
238 const size_t input_channels = fully_connected_op->group_input_channels; in setup_fully_connected_nc() local
301 size_t input_channels, in xnn_create_fully_connected_nc_f16()
366 size_t input_channels, in xnn_create_fully_connected_nc_f32()
427 size_t input_channels, in xnn_create_fully_connected_nc_qs8()
507 size_t input_channels, in xnn_create_fully_connected_nc_qu8()
H A Dtranspose-nd.c574 const size_t input_channels = output_channels * block_size * block_size; in xnn_create_depth_to_space_nchw2nhwc_x32() local
722 const size_t input_channels = output_channels * block_size * block_size; in create_depth_to_space_nhwc() local
931 size_t input_channels, in create_space_to_depth_nhwc()
1011 size_t input_channels, in xnn_create_space_to_depth_nhwc_x8()
1029 size_t input_channels, in xnn_create_space_to_depth_nhwc_x16()
1047 size_t input_channels, in xnn_create_space_to_depth_nhwc_x32()
/aosp_15_r20/external/mesa3d/src/gallium/drivers/etnaviv/
H A Detnaviv_ml_nn.c253 unsigned input_channels = operation->input_channels; in transpose() local
559 calc_addition_sizes(unsigned *input_width, unsigned *input_height, unsigned *input_channels, in calc_addition_sizes()
597 unsigned input_channels = operation->input_channels; in calculate_tiling() local
650 unsigned input_channels = operation->input_channels; in create_nn_config() local
971 unsigned input_channels = operation->addition ? 1 : operation->input_channels; in write_core_6() local
1051 unsigned input_channels = operation->addition ? 1 : operation->input_channels; in write_core_interleaved() local
1226 unsigned input_channels = operation->addition ? 1 : operation->input_channels; in calculate_weight_bo_size() local
1251 unsigned input_channels = operation->addition ? 1 : operation->input_channels; in calculate_zrl_bits() local
1303 unsigned input_channels = operation->addition ? 1 : operation->input_channels; in create_coefficients_bo() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_requantize_ops_test.cc118 const int input_channels = 2; in TEST_F() local
179 const int input_channels = 2; in TEST_F() local
237 const int input_channels = 2; in TEST_F() local
H A Dmkl_quantized_pooling_ops_test.cc54 const int input_channels = 2; in TEST_F() local
111 const int input_channels = 2; in TEST_F() local
/aosp_15_r20/external/mesa3d/src/gallium/targets/teflon/
H A Dtest_teflon.cpp30 std::vector<int> input_channels{1, 32, 120, 128, 256}; variable
148 test_conv(int input_size, int weight_size, int input_channels, int output_channels, in test_conv()
188 test_add(int input_size, int weight_size, int input_channels, int output_channels, in test_add()
488 int input_channels = atoi(argv[n++]); in main() local
H A Dtest_executor.cpp51 int input_channels, in patch_conv2d()
157 int input_channels, in conv2d_generate_model()
195 int input_channels, in add_generate_model()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_conv2d_acl.cc32 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in ACLDepthwiseConvImpl()
129 int64_t input_cols, int64_t input_channels, in ACLGemmConvImpl()
236 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in __xla_cpu_runtime_ACLConv2DF32()
H A Druntime_single_threaded_conv2d.cc25 int64_t input_cols, int64_t input_channels, int64_t kernel_rows, in __xla_cpu_runtime_EigenSingleThreadedConv2DF16()
45 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in __xla_cpu_runtime_EigenSingleThreadedConv2DF32()
/aosp_15_r20/external/pytorch/aten/src/ATen/benchmarks/
H A Dstateful_conv1d.cpp10 const size_t input_channels = static_cast<size_t>(state.range(0)); in stateful_conv1d() local
70 for (size_t input_channels = 32; input_channels < 256; input_channels *= 2) { in GenerateSizes() local
/aosp_15_r20/external/XNNPACK/bench/
H A Ddeconvolution.cc41 const size_t input_channels = state.range(11); in xnnpack_deconvolution_qu8() local
162 const size_t input_channels = state.range(11); in xnnpack_deconvolution_f32() local
280 const size_t input_channels = state.range(11); in tflite_deconvolution_f32() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_pooling_ops_test.cc55 const int input_channels = 2; in TEST_F() local
100 const int input_channels = 2; in TEST_F() local

12345