Home
last modified time | relevance | path

Searched refs:GpuKernelArgumentInfo (Results 1 – 20 of 20) sorted by relevance

/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/
DGpuKernelArgument.h37 struct GpuKernelArgumentInfo struct
59 GpuKernelArgumentInfo() = default; argument
61 GpuKernelArgumentInfo(Type type) in GpuKernelArgumentInfo() argument
68 bool operator==(const GpuKernelArgumentInfo &info0, const GpuKernelArgumentInfo &info1);
81 const GpuKernelArgumentInfo &kernel_arg_info) in GpuKernelArgument()
102 GpuKernelArgumentInfo *kernel_argument_info() in kernel_argument_info()
107 const GpuKernelArgumentInfo *kernel_argument_info() const in kernel_argument_info()
123 GpuKernelArgumentInfo _kernel_arg_info{};
DGpuWorkloadSourceCode.h57 const GpuKernelArgumentInfo &kernel_arg_info) in GpuWorkloadArgument()
89 GpuKernelArgumentInfo *kernel_argument_info() in kernel_argument_info()
94 const GpuKernelArgumentInfo *kernel_argument_info() const in kernel_argument_info()
111 GpuKernelArgumentInfo _kernel_arg_info{};
DGpuKernelArgument.cpp31 bool operator==(const GpuKernelArgumentInfo &info0, const GpuKernelArgumentInfo &info1) in operator ==()
/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/template_writer/cl/
DClTemplateDepthwiseConv2d.cpp203 const GpuKernelArgumentInfo::Type input_type = _settings.export_input_to_cl_image() ? in declare_variables()
204GpuKernelArgumentInfo::Type::Tensor_4D_t_Image : in declare_variables()
205GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer; in declare_variables()
210 GpuKernelArgumentInfo(input_type), in declare_variables()
213 const GpuKernelArgumentInfo::Type weight_type = _settings.export_weights_to_cl_image() ? in declare_variables()
214GpuKernelArgumentInfo::Type::Tensor_4D_t_Image : in declare_variables()
215GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer; in declare_variables()
220 GpuKernelArgumentInfo(weight_type), in declare_variables()
228 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Vector), in declare_variables()
234 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer), in declare_variables()
[all …]
DClTemplateDirectConv2d.cpp231 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer), in declare_variables()
234 …nst GpuKernelArgumentInfo::Type weight_type = _settings.export_to_cl_image() ? GpuKernelArgumentIn… in declare_variables()
238 GpuKernelArgumentInfo(weight_type), in declare_variables()
246 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Vector), in declare_variables()
252 GpuKernelArgumentInfo(common_tensor_type), in declare_variables()
282 case GpuKernelArgumentInfo::Type::Image_Export_To_ClImage2D: in get_tag_lut()
283 case GpuKernelArgumentInfo::Type::Image_3D_Export_To_ClImage2D: in get_tag_lut()
284 case GpuKernelArgumentInfo::Type::Tensor_4D_t_Image: in get_tag_lut()
DClTemplateLogits1DNorm.cpp106 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_3D), in declare_variables()
112 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_3D), in declare_variables()
118 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_3D), in declare_variables()
DClTemplateStore.cpp67 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer), in declare_variables()
72 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer), in declare_variables()
DClTemplateLogits1DMaxShiftExpSum.cpp197 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_3D), in declare_variables()
203 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_3D), in declare_variables()
209 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_3D), in declare_variables()
DClTemplateWriter.cpp248 case GpuKernelArgumentInfo::Type::Vector: in write_argument_declaration()
253 case GpuKernelArgumentInfo::Type::Image: in write_argument_declaration()
258 case GpuKernelArgumentInfo::Type::Image_3D: in write_argument_declaration()
264 case GpuKernelArgumentInfo::Type::Image_3D_Export_To_ClImage2D: in write_argument_declaration()
270 case GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer: in write_argument_declaration()
275 case GpuKernelArgumentInfo::Type::Tensor_4D_t_Image: in write_argument_declaration()
280 case GpuKernelArgumentInfo::Type::Tensor_3D: in write_argument_declaration()
DClTemplateActivation.cpp108 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer), in declare_variables()
114 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer), in declare_variables()
DClTemplateCast.cpp132 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer), in declare_variables()
138 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer), in declare_variables()
DClTemplateResize.cpp184 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer), in declare_variables()
190 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer), in declare_variables()
DClTemplatePool2d.cpp370 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer), in declare_variables()
376 GpuKernelArgumentInfo(GpuKernelArgumentInfo::Type::Tensor_4D_t_Buffer), in declare_variables()
DClTemplateElementwiseBinary.cpp153 GpuKernelArgumentInfo(common_tensor_type), in declare_variables()
159 GpuKernelArgumentInfo(common_tensor_type), in declare_variables()
165 GpuKernelArgumentInfo(common_tensor_type), in declare_variables()
DClTemplateReshape.cpp100 GpuKernelArgumentInfo(common_tensor_type), // GpuKernelArgumentInfo::Type::Image_3D in declare_variables()
106 GpuKernelArgumentInfo(common_tensor_type), in declare_variables()
/external/ComputeLibrary/src/dynamic_fusion/runtime/gpu/cl/
DClKernelRuntime.cpp60 inline void ClKernelRuntime::add_tensor_argument(unsigned int &idx, const GpuKernelArgumentInfo &ar… in add_tensor_argument()
64 case GpuKernelArgumentInfo::Type::Scalar: in add_tensor_argument()
70 case GpuKernelArgumentInfo::Type::Vector: in add_tensor_argument()
76 case GpuKernelArgumentInfo::Type::Image: in add_tensor_argument()
81 case GpuKernelArgumentInfo::Type::Image_Reinterpret_As_3D: in add_tensor_argument()
88 case GpuKernelArgumentInfo::Type::Image_Export_To_ClImage2D: in add_tensor_argument()
98 case GpuKernelArgumentInfo::Type::Image_3D: in add_tensor_argument()
104 case GpuKernelArgumentInfo::Type::Image_3D_Export_To_ClImage2D: in add_tensor_argument()
115 case GpuKernelArgumentInfo::Type::Tensor_3D: in add_tensor_argument()
121 case GpuKernelArgumentInfo::Type::Tensor_4D: in add_tensor_argument()
[all …]
DClKernelRuntime.h68 …inline void add_tensor_argument(unsigned int &idx, const GpuKernelArgumentInfo &arg, const ICLTens…
/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/template_writer/
DIGpuTemplateComponentWriter.h52 …static constexpr GpuKernelArgumentInfo::Type common_tensor_type = GpuKernelArgumentInfo::Type::Ten…
DGpuKernelVariableTable.h62 GpuKernelArgumentInfo kernel_argument_info{};
78 …t GpuKernelComponentGroup &comp_group, const ITensorInfo *tensor, GpuKernelArgumentInfo argument_i…
DGpuKernelVariableTable.cpp35 …t GpuKernelComponentGroup &comp_group, const ITensorInfo *tensor, GpuKernelArgumentInfo argument_i… in declare_variable()