Home
last modified time | relevance | path

Searched refs:ukernel_type (Results 1 – 12 of 12) sorted by relevance

/external/XNNPACK/src/
Dconvolution-nhwc.c232 enum xnn_ukernel_type ukernel_type = xnn_ukernel_type_none; in xnn_create_convolution2d_nhwc_q8() local
237 ukernel_type = xnn_ukernel_type_dwconv; in xnn_create_convolution2d_nhwc_q8()
239 ukernel_type = xnn_ukernel_type_gemm; in xnn_create_convolution2d_nhwc_q8()
241 ukernel_type = xnn_ukernel_type_igemm; in xnn_create_convolution2d_nhwc_q8()
245 switch (ukernel_type) { in xnn_create_convolution2d_nhwc_q8()
299 switch (ukernel_type) { in xnn_create_convolution2d_nhwc_q8()
381 convolution_op->ukernel.type = ukernel_type; in xnn_create_convolution2d_nhwc_q8()
541 enum xnn_ukernel_type ukernel_type = xnn_ukernel_type_none; in xnn_create_convolution2d_nhwc_f32() local
545 ukernel_type = xnn_ukernel_type_vmulcaddc; in xnn_create_convolution2d_nhwc_f32()
549 ukernel_type = xnn_ukernel_type_dwconv; in xnn_create_convolution2d_nhwc_f32()
[all …]
Dconvolution-nchw.c143 enum xnn_ukernel_type ukernel_type; in xnn_create_convolution2d_nchw_f32() local
158 ukernel_type = xnn_ukernel_type_spmm; in xnn_create_convolution2d_nchw_f32()
163 ukernel_type = xnn_ukernel_type_dconv2d_hwc2spchw; in xnn_create_convolution2d_nchw_f32()
168 ukernel_type = xnn_ukernel_type_dwconv; in xnn_create_convolution2d_nchw_f32()
174 ukernel_type = xnn_ukernel_type_dwconv; in xnn_create_convolution2d_nchw_f32()
180 ukernel_type = xnn_ukernel_type_dwconv; in xnn_create_convolution2d_nchw_f32()
186 ukernel_type = xnn_ukernel_type_dwconv; in xnn_create_convolution2d_nchw_f32()
202 switch (ukernel_type) { in xnn_create_convolution2d_nchw_f32()
452 if (ukernel_type == xnn_ukernel_type_dwconv) { in xnn_create_convolution2d_nchw_f32()
459 convolution_op->ukernel.type = ukernel_type; in xnn_create_convolution2d_nchw_f32()
Ddeconvolution-nhwc.c198 enum xnn_ukernel_type ukernel_type = xnn_ukernel_type_igemm; in xnn_create_deconvolution2d_nhwc_q8() local
201 ukernel_type = xnn_ukernel_type_subconv2d; in xnn_create_deconvolution2d_nhwc_q8()
233 switch (ukernel_type) { in xnn_create_deconvolution2d_nhwc_q8()
287 deconvolution_op->ukernel.type = ukernel_type; in xnn_create_deconvolution2d_nhwc_q8()
447 enum xnn_ukernel_type ukernel_type = xnn_ukernel_type_igemm; in xnn_create_deconvolution2d_nhwc_f32() local
450 ukernel_type = xnn_ukernel_type_subconv2d; in xnn_create_deconvolution2d_nhwc_f32()
482 switch (ukernel_type) { in xnn_create_deconvolution2d_nhwc_f32()
528 deconvolution_op->ukernel.type = ukernel_type; in xnn_create_deconvolution2d_nhwc_f32()
/external/XNNPACK/tools/
Dgenerate-pack-test.py170 _, datatype, ukernel_type, _ = ukernel.split("_", 3)
174 "UKERNEL_TYPE": ukernel_type.upper(),
Dgenerate-bilinear-test.py168 _, datatype, ukernel_type, _ = ukernel.split("_", 3)
173 "UKERNEL_TYPE": ukernel_type.upper(),
Dgenerate-spmm-test.py381 _, datatype, ukernel_type, _ = ukernel.split("_", 3)
388 "UKERNEL_TYPE": ukernel_type.upper(),
Dgenerate-dwconv-test.py274 _, datatype, ukernel_type, _ = ukernel.split("_", 3)
281 "UKERNEL_TYPE": ukernel_type.upper(),
Dgenerate-gemm-test.py783 _, datatype, ukernel_type, _ = ukernel.split("_", 3)
790 "UKERNEL_TYPE": ukernel_type.upper(),
Dgenerate-maxpool-test.py1021 _, datatype, ukernel_type, _ = ukernel.split("_", 3)
Dgenerate-prelu-test.py216 _, datatype, ukernel_type, _ = ukernel.split("_", 3)
Dgenerate-vmulcaddc-test.py224 _, datatype, ukernel_type, _ = ukernel.split("_", 3)
Dgenerate-argmaxpool-test.py1034 _, datatype, ukernel_type, _ = ukernel.split("_", 3)