Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/
H A Dsoftmax.h129 TensorT row_max, row_sum; member
/aosp_15_r20/external/webrtc/modules/rtp_rtcp/test/testFec/
H A Dtest_fec.cc227 uint32_t row_sum = 0; in RunTest() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/
DQuantUtils.cpp56 int32_t row_sum = 0; in MatrixScalarMultiplyAccumulate() local
/aosp_15_r20/external/harfbuzz_ng/util/
H A Dansi-print.hh225 unsigned int row_sum[CELL_H] = {0}; in block_best() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dbatch_matmul.h188 int32_t row_sum = woff_ptr2[i]; in BatchMatMul() local
H A Dportable_tensor_utils.cc544 int32_t row_sum = 0; in PortableMatrixScalarMultiplyAccumulate() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp224 int64_t row_sum = 0; in hsum() local
281 int64_t row_sum = 0; in hsum() local
338 int64_t row_sum = 0; in hsum() local
393 int64_t row_sum = 0; in hsum_sq() local
467 int64_t row_sum = 0; in hsum_sq() local
549 float row_sum = 0; in hsum_sq() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
H A Dquantized.cpp87 int32_t row_sum = row_bias[row]; in requantize_block_32_int() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/test/
H A Dgemm-microkernel-tester.h651 int32_t* row_sum, in q8gemm_compute_row_sum()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/bench/
H A Dq8gemm.cc600 int32_t* row_sum) { in q8gemm_compute_row_sum()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dsse_tensor_utils.cc646 int32_t row_sum = ReduceInt32x4(row_sum_32x4); in SseReductionSumVector() local
H A Dneon_tensor_utils.cc718 int64_t row_sum = 0; in DotprodSparseMatrixBatchVectorMultiplyAccumulate() local
1251 int32x4_t row_sum = vdupq_n_s32(0); in NeonMatrixScalarMultiplyAccumulate() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DSumKernel.cpp417 scalar_t row_sum(const char * C10_RESTRICT in_data, in row_sum() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dfully_connected.cc758 int32_t row_sum = 0; in EvalHybrid() local
/aosp_15_r20/external/libvpx/third_party/libyuv/source/
H A Drow_common.cc2502 int32_t row_sum[4] = {0, 0, 0, 0}; in ComputeCumulativeSumRow_C() local
/aosp_15_r20/external/libaom/third_party/libyuv/source/
H A Drow_common.cc2905 int32_t row_sum[4] = {0, 0, 0, 0}; in ComputeCumulativeSumRow_C() local
/aosp_15_r20/external/libyuv/source/
H A Drow_common.cc3571 int32_t row_sum[4] = {0, 0, 0, 0}; in ComputeCumulativeSumRow_C() local