Home
last modified time | relevance | path

Searched defs:accumulation_buffer (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dwhile_loop_all_reduce_code_motion_test.cc104 HloInstruction* accumulation_buffer = in TEST_F() local
398 HloInstruction* accumulation_buffer = in TEST_F() local
488 HloInstruction* accumulation_buffer = in TEST_F() local
659 HloInstruction* accumulation_buffer = in TEST_F() local
740 HloInstruction* accumulation_buffer = in TEST_F() local
821 HloInstruction* accumulation_buffer = in TEST_F() local
H A Dwhile_loop_all_reduce_code_motion.cc48 HloInstruction* accumulation_buffer; member
502 HloInstruction* accumulation_buffer = in IsAllReduceMovable() local
620 HloInstruction* accumulation_buffer = in CreateSinkedAllReduces() local
/aosp_15_r20/external/XNNPACK/src/f32-argmaxpool/
H A D9p8x-scalar-c1.c18 float* accumulation_buffer, in xnn_f32_argmaxpool_ukernel_9p8x__scalar_c1()
H A D9p8x-wasmsimd-c4.c19 float* accumulation_buffer, in xnn_f32_argmaxpool_ukernel_9p8x__wasmsimd_c4()
H A D9p8x-neon-c4.c19 float* accumulation_buffer, in xnn_f32_argmaxpool_ukernel_9p8x__neon_c4()
H A D9p8x-sse2-c4.c19 float* accumulation_buffer, in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dlstm_eval.cc204 float* accumulation_buffer = gate; in CalculateLstmGateFloat() local
/aosp_15_r20/external/XNNPACK/src/amalgam/
H A Dsse2.c430 float* accumulation_buffer, in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4()