Home
last modified time | relevance | path

Searched defs:gpu_target (Results 1 – 20 of 20) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/
H A DClGemm.cpp214 const GPUTarget gpu_target = CLScheduler::get().target(); in configure_native() local
246 const GPUTarget gpu_target = CLScheduler::get().target(); in configure_reshaped() local
291 const GPUTarget gpu_target = CLScheduler::get().target(); in configure_reshaped_only_rhs() local
338 const GPUTarget gpu_target = CLScheduler::get().target(); in configure_reshaped_only_rhs_mmul() local
381 const GPUTarget gpu_target = CLScheduler::get().target(); in validate_native() local
418 const GPUTarget gpu_target = CLScheduler::get().target(); in validate_reshaped() local
467 const GPUTarget gpu_target = CLScheduler::get().target(); in validate_reshaped_only_rhs() local
516 const GPUTarget gpu_target = CLScheduler::get().target(); in validate_reshaped_only_rhs_mmul() local
H A DClConv2d.cpp49 size_t get_direct_conv_kernel_threshold_nhwc(arm_compute::GPUTarget gpu_target) in get_direct_conv_kernel_threshold_nhwc()
137 const GPUTarget gpu_target = CLScheduler::get().target(); in validate() local
180 … const WeightsInfo &weights_info, const GPUTarget gpu_target) in get_convolution_method()
H A DClGemmLowpMatrixMultiplyCore.cpp272 const GPUTarget gpu_target = CLScheduler::get().target(); in configure() local
492 const GPUTarget gpu_target = CLScheduler::get().target(); in validate() local
H A DClDirectConv2d.cpp59 GPUTarget gpu_target = CLScheduler::get().target(); in config_direct_convolution_nhwc() local
H A DClIndirectConv2d.cpp53 GPUTarget gpu_target = CLScheduler::get().target(); in config_indirect_convolution_nhwc() local
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/operators/
H A DGpuConv2d.cpp48 bool export_to_cl_image_support(const ITensorInfo *tensor, GPUTarget gpu_target, const cl::Device &… in export_to_cl_image_support()
94 GPUTarget gpu_target = CLScheduler::get().target(); in config_direct_convolution_nhwc() local
144 const auto gpu_target = context.gpu_target(); in is_supported_op_helper() local
255 const auto gpu_target = sketch_ctx->gpu_target(); in create_op() local
H A DGpuDepthwiseConv2d.cpp92 const GpuTarget gpu_target = context.gpu_target(); in is_supported_op_helper() local
204 const GpuTarget gpu_target = sketch_ctx->gpu_target(); in create_op() local
/aosp_15_r20/external/ComputeLibrary/tests/CL/
H A DHelper.h66 void configure(GPUTarget gpu_target, Args &&... args) in configure()
146 void configure(GPUTarget gpu_target, Args &&... args) in configure()
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLDepthwiseConvolutionLayer.cpp94 const GPUTarget gpu_target = CLScheduler::get().target(); in configure() local
177 const GPUTarget gpu_target = CLScheduler::get().target(); in validate() local
H A DCLConvolutionLayer.cpp135 const GPUTarget gpu_target = CLScheduler::get().target(); in validate() local
164 …eightsInfo &weights_info, const ActivationLayerInfo &act_info, const GPUTarget gpu_target, const S… in get_convolution_method()
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/
H A DGpuWorkloadContext.cpp38 GpuTarget GpuWorkloadContext::gpu_target() const in gpu_target() function in arm_compute::experimental::dynamic_fusion::GpuWorkloadContext
/aosp_15_r20/external/ComputeLibrary/src/core/
H A DGPUTarget.cpp205 GPUTarget gpu_target; in get_target_from_name() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/gemm_auto_heuristics/
H A DCLGEMMAutoHeuristics.h42 GPUTarget gpu_target; /**< Which @ref GPUTarget to query about */ member
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClIndirectConv2dKernel.cpp132 const GPUTarget gpu_target = get_target(); in configure() local
H A DClWinogradOutputTransformKernel.cpp182 const GPUTarget gpu_target = get_target(); in configure() local
H A DClDirectConv2dKernel.cpp172 const GPUTarget gpu_target = get_target(); in configure() local
/aosp_15_r20/external/ComputeLibrary/src/core/CL/
H A DCLHelpers.cpp244 const GPUTarget gpu_target = get_target_from_name(device_name); in dot8_supported() local
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLDepthwiseConvolutionLayerNativeKernel.cpp226 const GPUTarget gpu_target = get_target(); in configure() local
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dconfig.py1135 gpu_target = "host-cuda" variable in halide
/aosp_15_r20/external/ComputeLibrary/utils/
H A DTypePrinter.h2432 inline std::string to_string(const GPUTarget &gpu_target) in to_string()