Searched defs:out_accum (Results 1 – 3 of 3) sorted by relevance
| /external/libaom/av1/encoder/x86/ |
| D | cnn_avx2.c | 186 const float *input_ptr, int in_stride, __m256 *weight, __m256 *out_accum, in perform_convolve_for_8h_2x2_blocks()
213 const float *input_ptr, int in_stride, __m256 *weight, __m256 *out_accum, in perform_convolve_for_4hx2v_2x2_blocks()
387 __m256 out_accum[8]; in cnn_convolve_no_maxpool_padding_valid_layer1_avx2() local
434 __m256 out_accum[2]; in cnn_convolve_no_maxpool_padding_valid_layer2_avx2() local
|
| /external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/ |
| D | flash_api.cpp | 310 at::Tensor out_accum; in set_params_splitkv() local
483 at::Tensor softmax_lse_accum, out_accum; in mha_fwd() local
733 at::Tensor softmax_lse_accum, out_accum; in mha_varlen_fwd() local
1490 at::Tensor softmax_lse_accum, out_accum; in mha_fwd_kvcache() local
|
| /external/executorch/kernels/test/ |
| D | op_index_put_test.cpp | 220 Tensor out_accum = tf.ones(out_size); in run_test_cases() local
|