/external/XNNPACK/src/operators/ |
D | argmax-pooling-nhwc.c | 34 const struct argmaxpool_parameters* ukernel) in select_ukernel() 221 …const struct argmaxpool_parameters* ukernel = select_ukernel(pooling_size, xnn_params.f32.argmaxpo… in xnn_setup_argmax_pooling2d_nhwc_f32() local
|
/external/XNNPACK/tools/ |
D | generate-raddextexp-test.py | 81 def generate_test_cases(ukernel, elements_tile, isa): argument
|
D | generate-vscaleexpminusmax-test.py | 96 def generate_test_cases(ukernel, elements_tile, isa): argument
|
D | generate-raddexpminusmax-test.py | 81 def generate_test_cases(ukernel, elements_tile, isa): argument
|
D | generate-clamp-test.py | 120 def generate_test_cases(ukernel, batch_tile, isa): argument
|
D | generate-hswish-test.py | 92 def generate_test_cases(ukernel, batch_tile, isa): argument
|
D | generate-vscaleextexp-test.py | 81 def generate_test_cases(ukernel, elements_tile, isa): argument
|
D | generate-raddstoreexpminusmax-test.py | 81 def generate_test_cases(ukernel, elements_tile, isa): argument
|
D | generate-ibilinear-chw-test.py | 152 def generate_test_cases(ukernel, channel_tile, pixel_tile, isa): argument
|
D | generate-pack-test.py | 155 def generate_test_cases(ukernel, mr, k_block, isa): argument
|
D | generate-vmulcaddc-test.py | 208 def generate_test_cases(ukernel, channel_tile, row_tile, isa): argument
|
D | generate-vunary-test.py | 160 def generate_test_cases(ukernel, op_type, batch_tile, isa): argument
|
D | generate-ibilinear-test.py | 153 def generate_test_cases(ukernel, channel_tile, pixel_tile, isa): argument
|
D | generate-prelu-test.py | 170 def generate_test_cases(ukernel, row_tile, channel_tile, isa): argument
|
D | generate-avgpool-test.py | 2395 def generate_test_cases(ukernel, primary_tile, incremental_tile, channel_tile, argument
|
D | generate-argmaxpool-test.py | 663 def generate_test_cases(ukernel, primary_tile, incremental_tile, channel_tile, argument
|
D | generate-maxpool-test.py | 1002 def generate_test_cases(ukernel, primary_tile, incremental_tile, channel_tile, argument
|
D | generate-gavgpool-test.py | 509 def generate_test_cases(ukernel, primary_tile, incremental_tile, channel_tile, argument
|
D | generate-dwconv-test.py | 285 def generate_test_cases(ukernel, cr, kr, c_block, is_pipelined, isa): argument
|
D | generate-vbinary-test.py | 257 def generate_test_cases(ukernel, op_type, activation_type, tester, batch_tile, isa): argument
|
D | generate-spmm-test.py | 379 def generate_test_cases(ukernel, mr, nr, k_block, is_pipelined, isa): argument
|
D | generate-gemm-test.py | 761 def generate_test_cases(ukernel, mr, nr, kr, sr, argument
|
D | generate-dwconv2d-chw-test.py | 243 def generate_test_cases(ukernel, kernel_height, kernel_width, subsampling, \ argument
|
/external/XNNPACK/src/xnnpack/ |
D | compute.h | 76 struct xnn_hmp_gemm_ukernel ukernel; member 146 xnn_spmm_ukernel_function ukernel; member 179 struct xnn_hmp_igemm_ukernel ukernel; member 275 struct xnn_hmp_gemm_ukernel ukernel; member 321 struct xnn_hmp_igemm_ukernel ukernel; member 452 xnn_univector_ukernel_function ukernel; member 480 xnn_depthtospace2d_chw2hwc_ukernel_function ukernel; member 506 xnn_maxpool_ukernel_function ukernel; member 529 xnn_unpool_ukernel_function ukernel; member 686 xnn_gavgpool_cw_ukernel_function ukernel; member [all …]
|
D | params.h | 1744 struct xnn_hmp_gemm_ukernel ukernel = { function }; in xnn_init_hmp_gemm_ukernel() local 1751 static inline bool xnn_is_hmp_gemm_ukernel(struct xnn_hmp_gemm_ukernel ukernel) { in xnn_is_hmp_gemm_ukernel() 1769 struct xnn_hmp_igemm_ukernel ukernel = { function }; in xnn_init_hmp_igemm_ukernel() local 1776 static inline bool xnn_is_hmp_igemm_ukernel(struct xnn_hmp_igemm_ukernel ukernel) { in xnn_is_hmp_igemm_ukernel() 1822 xnn_spmm_ukernel_function ukernel; member 1844 xnn_dwconv2d_chw_ukernel_function ukernel; member 1853 xnn_gavgpool_cw_ukernel_function ukernel; member 1873 xnn_depthtospace2d_chw2hwc_ukernel_function ukernel; member 1912 xnn_maxpool_ukernel_function ukernel; member 1918 xnn_ibilinear_ukernel_function ukernel; member [all …]
|