Searched defs:output_channel_index (Results 1 – 17 of 17) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/ |
| H A D | requantization.h | 468 int32_t output_channel_index) { in pytorch_qnnp_fp32_requantize()
490 int32_t output_channel_index) { in pytorch_qnnp_fp32_requantize_magic()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/q8conv/ |
| H A D | 4x4c2-sse2.c | 23 size_t output_channel_index, in pytorch_q8conv_ukernel_4x4c2__sse2()
|
| H A D | 4x8-neon.c | 23 size_t output_channel_index, in pytorch_q8conv_ukernel_4x8__neon()
|
| H A D | 8x8-neon.c | 23 size_t output_channel_index, in pytorch_q8conv_ukernel_8x8__neon()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm/ |
| H A D | 4x4c2-dq-sse2.c | 24 size_t output_channel_index, in pytorch_q8gemm_dq_ukernel_4x4c2__sse2()
|
| H A D | 4x4c2-sse2.c | 23 size_t output_channel_index, in pytorch_q8gemm_ukernel_4x4c2__sse2()
|
| H A D | 2x4c8-sse2.c | 50 size_t output_channel_index, in pytorch_q8gemm_ukernel_2x4c8__sse2()
|
| H A D | 4x8-dq-neon.c | 24 size_t output_channel_index, in pytorch_q8gemm_dq_ukernel_4x8__neon()
|
| H A D | 6x4-neon.c | 23 size_t output_channel_index, in pytorch_q8gemm_ukernel_6x4__neon()
|
| H A D | 4x8-neon.c | 23 size_t output_channel_index, in pytorch_q8gemm_ukernel_4x8__neon()
|
| H A D | 8x8-neon.c | 23 size_t output_channel_index, in pytorch_q8gemm_ukernel_8x8__neon()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | fc-run.cc | 41 size_t output_channel_index = nr_block_start; in compute_q8gemm() local
|
| H A D | operator-run.c | 59 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
|
| H A D | conv-run.cc | 100 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
|
| H A D | fc-dynamic-run.cc | 42 size_t output_channel_index = nr_block_start; in compute_q8gemm_dq() local
|
| H A D | deconv-run.cc | 67 const size_t output_channel_index = group_index * n + nr_block_start; in compute_q8conv() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/ |
| H A D | 8x4c1x4-dq-packedA-sse2.h | 42 size_t output_channel_index, in KERNEL_NAME()
|