Searched defs:accumulation_buffer (Results 1 – 6 of 6) sorted by relevance
103 HloInstruction* accumulation_buffer = in TEST_F() local397 HloInstruction* accumulation_buffer = in TEST_F() local568 HloInstruction* accumulation_buffer = in TEST_F() local649 HloInstruction* accumulation_buffer = in TEST_F() local
49 HloInstruction* accumulation_buffer; member307 HloInstruction* accumulation_buffer = in IsAllReduceMovable() local412 HloInstruction* accumulation_buffer = in CreateSinkedAllReduces() local
18 float* accumulation_buffer, in xnn_f32_argmaxpool_ukernel_9p8x__scalar_c1()
19 float* accumulation_buffer, in xnn_f32_argmaxpool_ukernel_9p8x__neon_c4()
19 float* accumulation_buffer, in xnn_f32_argmaxpool_ukernel_9p8x__wasmsimd_c4()
19 float* accumulation_buffer, in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4()