| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | DeconvolutionLayerFixture.h | 251 unsigned int num_kernels, DataType data_type, DataLayout data_layout, bool add_bias) in setup()
272 …unsigned int pad_bottom, unsigned int num_kernels, DataType data_type, DataLayout data_layout, boo… in setup()
293 …unsigned int num_kernels, DataType data_type, DataLayout data_layout, QuantizationInfo input_quant… in setup()
315 …unsigned int num_kernels, DataType data_type, DataLayout data_layout, QuantizationInfo input_quant… in setup()
|
| H A D | DirectConvolution3DFixture.h | 162 …unsigned int num_kernels, bool has_bias, ActivationLayerInfo act_info, DataType data_type, DataLay… in setup()
175 …unsigned int num_kernels, bool has_bias, ActivationLayerInfo act_info, DataType data_type, DataLay… in setup()
|
| H A D | DirectConvolutionLayerFixture.h | 234 …, int stride_y, int pad_x, int pad_y, unsigned int kernel_size, unsigned int num_kernels, DataType… in setup()
247 …, int stride_y, int pad_x, int pad_y, unsigned int kernel_size, unsigned int num_kernels, DataType… in setup()
|
| H A D | ConvolutionLayerFixture.h | 580 …const int num_kernels = weights_shape[get_data_layout_dimension_index(_data_layout, DataLayoutDi… in compute_target() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | im2col.cuh | 84 int64_t num_kernels = channels * height_col * width_col; in im2col() local
217 int64_t num_kernels = channels * height * width; in col2im() local
312 const int64_t num_kernels = channels * height * width; in col2im_batched() local
|
| H A D | UpSampleTrilinear3d.cu | 125 const int num_kernels, in upsample_trilinear3d_backward_out_frame()
262 const int num_kernels = output_depth * output_height * output_width; in upsample_trilinear3d_out_cuda_template() local
330 const int num_kernels = output_depth * output_height * output_width; in upsample_trilinear3d_backward_out_cuda_template() local
|
| H A D | UpSampleLinear1d.cu | 136 const int num_kernels = output_width; in upsample_linear1d_out_cuda_template() local
180 const int num_kernels = output_width; in upsample_linear1d_backward_out_cuda_template() local
|
| H A D | ForeachReduceOp.cu | 192 const size_t num_kernels = ceil_div(ntensors, MAX_TENSORS_PER_KERNEL); in foreach_tensor_max_cuda() local
502 const size_t num_kernels = in foreach_tensor_norm_cuda() local
|
| H A D | vol2col.cuh | 101 const auto num_kernels = static_cast<int64_t>(channels) * depth_col * height_col * width_col; in vol2col() local
222 const auto num_kernels = channels * depth * height * width; in col2vol() local
|
| H A D | UpSampleNearest2d.cu | 247 const int64_t num_kernels = output.numel(); in upsample_nearest2d_out_cuda_template() local
377 const int num_kernels = grad_input.numel(); in upsample_nearest2d_backward_out_cuda_template() local
|
| H A D | UpSampleBilinear2d.cu | 306 const int num_kernels = output.numel(); in upsample_bilinear2d_out_cuda_template() local
333 const int num_kernels = output_height * output_width; in upsample_bilinear2d_out_cuda_template() local
390 const size_t num_kernels = nbatch * channels * output_height * output_width; in upsample_bilinear2d_backward_out_cuda_template() local
|
| H A D | UpSampleBicubic2d.cu | 244 const int num_kernels = output_height * output_width; in upsample_bicubic2d_backward_out_cuda_template() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/ |
| H A D | DirectConv2dFixture.h | 226 …t stride_x, int stride_y, int pad_x, int pad_y, unsigned int kernel_size, unsigned int num_kernels, in setup()
345 …, int stride_y, int pad_x, int pad_y, unsigned int kernel_size, unsigned int num_kernels, DataType… in setup()
|
| /aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/ |
| H A D | ClGemmConv2d.cpp | 180 const unsigned int num_kernels = weights->dimension(idx_kernels); in configure() local
375 const unsigned int num_kernels = weights->dimension(idx_kernels); in validate() local
|
| /aosp_15_r20/external/OpenCL-CTS/test_conformance/basic/ |
| H A D | test_createkernelsinprogram.cpp | 58 unsigned int num_kernels; in test_createkernelsinprogram() local
|
| /aosp_15_r20/external/intel-media-driver/media_softlet/agnostic/common/vp/cm_fc_ld/ |
| H A D | cm_fc_ld.cpp | 73 int cm_fc_combine_kernels(size_t num_kernels, cm_fc_kernel_t kernels[], in cm_fc_combine_kernels()
|
| /aosp_15_r20/external/vboot_reference/cgpt/ |
| H A D | cgpt_prioritize.c | 103 int num_kernels; in CgptPrioritize() local
|
| /aosp_15_r20/external/mesa3d/src/compiler/clc/ |
| H A D | clc_helpers.cpp | 693 unsigned *num_kernels,
757 unsigned num_kernels)
|
| H A D | clc.h | 190 unsigned num_kernels; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | pooling_ops_3d_gpu.cu.cc | 145 int num_kernels = in operator ()() local
|
| H A D | maxpooling_op_gpu.cu.cc | 431 const int num_kernels = batch * channels * pooled_height * pooled_width; in operator ()() local
|
| /aosp_15_r20/external/OpenCL-CTS/test_conformance/spir/ |
| H A D | run_build_test.cpp | 207 size_t num_kernels; in process() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/frontends/rusticl/api/ |
| H A D | kernel.rs | 316 let mut num_kernels = 0; in create_kernels_in_program() localVariable
|
| /aosp_15_r20/external/OpenCL-CTS/test_conformance/vulkan/ |
| H A D | test_vulkan_interop_image.cpp | 1286 const uint32_t num_kernels = ARRAY_SIZE(num2DImagesList) + 1; in test_image_common() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/runtime/ |
| H A D | triton_heuristics.py | 1740 *numels, num_kernels, min_blocks, is_sequential, default_meta=None argument
|