Home
last modified time | relevance | path

Searched defs:tmp_max (Results 1 – 4 of 4) sorted by relevance

/external/mesa3d/src/mesa/vbo/
Dvbo_minmax_index.c383 unsigned tmp_min, tmp_max; in vbo_get_minmax_indices_gallium() local
/external/executorch/extension/llm/custom_ops/
Dop_sdpa.cpp128 scalar_t tmp_max = -std::numeric_limits<scalar_t>::infinity(); in _mul_reduce_max_fusion_kernel() local
516 accum_t tmp_max = 0, tmp_sum = 0, exp_tmp = 0; in cpu_flash_attention() local
/external/tensorflow/tensorflow/core/kernels/
Dquantized_instance_norm.cc179 float32x2_t tmp_max = vpmax_f32(low, high); in MinAndMax() local
/external/pytorch/aten/src/ATen/native/cpu/
DFlashAttentionKernel.cpp97 scalar_t tmp_max = -std::numeric_limits<scalar_t>::infinity(); in _mul_reduce_max_fusion_kernel() local
336 accum_t tmp_max = 0, tmp_sum = 0, exp_tmp = 0; in cpu_flash_attention() local