/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/ |
H A D | softmax.h | 129 TensorT row_max, row_sum; member
|
/aosp_15_r20/external/webrtc/modules/rtp_rtcp/test/testFec/ |
H A D | test_fec.cc | 227 uint32_t row_sum = 0; in RunTest() local
|
/aosp_15_r20/packages/modules/NeuralNetworks/common/ |
D | QuantUtils.cpp | 56 int32_t row_sum = 0; in MatrixScalarMultiplyAccumulate() local
|
/aosp_15_r20/external/harfbuzz_ng/util/ |
H A D | ansi-print.hh | 225 unsigned int row_sum[CELL_H] = {0}; in block_best() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
H A D | batch_matmul.h | 188 int32_t row_sum = woff_ptr2[i]; in BatchMatMul() local
|
H A D | portable_tensor_utils.cc | 544 int32_t row_sum = 0; in PortableMatrixScalarMultiplyAccumulate() local
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
H A D | QuantizedOpKernels.cpp | 224 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 D | quantized.cpp | 87 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 D | gemm-microkernel-tester.h | 651 int32_t* row_sum, in q8gemm_compute_row_sum()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/bench/ |
H A D | q8gemm.cc | 600 int32_t* row_sum) { in q8gemm_compute_row_sum()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
H A D | sse_tensor_utils.cc | 646 int32_t row_sum = ReduceInt32x4(row_sum_32x4); in SseReductionSumVector() local
|
H A D | neon_tensor_utils.cc | 718 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 D | SumKernel.cpp | 417 scalar_t row_sum(const char * C10_RESTRICT in_data, in row_sum() function
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | fully_connected.cc | 758 int32_t row_sum = 0; in EvalHybrid() local
|
/aosp_15_r20/external/libvpx/third_party/libyuv/source/ |
H A D | row_common.cc | 2502 int32_t row_sum[4] = {0, 0, 0, 0}; in ComputeCumulativeSumRow_C() local
|
/aosp_15_r20/external/libaom/third_party/libyuv/source/ |
H A D | row_common.cc | 2905 int32_t row_sum[4] = {0, 0, 0, 0}; in ComputeCumulativeSumRow_C() local
|
/aosp_15_r20/external/libyuv/source/ |
H A D | row_common.cc | 3571 int32_t row_sum[4] = {0, 0, 0, 0}; in ComputeCumulativeSumRow_C() local
|