Home
last modified time | relevance | path

Searched defs:output_channel_index (Results 1 – 17 of 17) sorted by relevance

/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/
Drequantization.h468 int32_t output_channel_index) { in pytorch_qnnp_fp32_requantize()
490 int32_t output_channel_index) { in pytorch_qnnp_fp32_requantize_magic()
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/q8conv/
D4x4c2-sse2.c23 size_t output_channel_index, in pytorch_q8conv_ukernel_4x4c2__sse2()
D4x8-neon.c23 size_t output_channel_index, in pytorch_q8conv_ukernel_4x8__neon()
D8x8-neon.c23 size_t output_channel_index, in pytorch_q8conv_ukernel_8x8__neon()
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
Dfc-run.cc41 size_t output_channel_index = nr_block_start; in compute_q8gemm() local
Dfc-dynamic-run.cc42 size_t output_channel_index = nr_block_start; in compute_q8gemm_dq() local
Ddeconv-run.cc67 const size_t output_channel_index = group_index * n + nr_block_start; in compute_q8conv() local
Dconv-run.cc100 const size_t output_channel_index = nr_block_start + group_index * n; in compute_q8gemm() local
154 const size_t output_channel_index = group_index * n + nr_block_start; in compute_q8conv() local
Doperator-run.c59 size_t output_channel_index = nr_block_start + group_index * n; in compute_q8gemm() local
106 size_t output_channel_index = nr_block_start; in compute_q8gemm_sparse_dq() local
198 const size_t output_channel_index = nr_block_start; in compute_q8gemm_prepacked_sparse_dq() local
390 size_t output_channel_index = nr_block_start + group_index * n; in compute_q8conv() local
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/
D4x4c2-dq-sse2.c24 size_t output_channel_index, in pytorch_q8gemm_dq_ukernel_4x4c2__sse2()
D4x4c2-sse2.c23 size_t output_channel_index, in pytorch_q8gemm_ukernel_4x4c2__sse2()
D2x4c8-sse2.c50 size_t output_channel_index, in pytorch_q8gemm_ukernel_2x4c8__sse2()
D4x8-dq-neon.c24 size_t output_channel_index, in pytorch_q8gemm_dq_ukernel_4x8__neon()
D6x4-neon.c23 size_t output_channel_index, in pytorch_q8gemm_ukernel_6x4__neon()
D4x8-neon.c23 size_t output_channel_index, in pytorch_q8gemm_ukernel_4x8__neon()
D8x8-neon.c23 size_t output_channel_index, in pytorch_q8gemm_ukernel_8x8__neon()
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/
D8x4c1x4-dq-packedA-sse2.h42 size_t output_channel_index, in KERNEL_NAME()