Home
last modified time | relevance | path

Searched refs:accumulation_buffer (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_all_reduce_code_motion_test.cc104 HloInstruction* accumulation_buffer = in TEST_F() local
106 EXPECT_THAT(accumulation_buffer, op::Constant()); in TEST_F()
398 HloInstruction* accumulation_buffer = in TEST_F() local
400 EXPECT_THAT(accumulation_buffer, op::Constant()); in TEST_F()
488 HloInstruction* accumulation_buffer = in TEST_F() local
490 EXPECT_THAT(accumulation_buffer, op::Constant()); in TEST_F()
659 HloInstruction* accumulation_buffer = in TEST_F() local
661 EXPECT_THAT(accumulation_buffer, op::Constant()); in TEST_F()
740 HloInstruction* accumulation_buffer = in TEST_F() local
742 EXPECT_THAT(accumulation_buffer, op::Constant()); in TEST_F()
[all …]
Dwhile_loop_all_reduce_code_motion.cc48 HloInstruction* accumulation_buffer; member
502 HloInstruction* accumulation_buffer = in IsAllReduceMovable() local
506 get_origin_tuple_index(accumulation_buffer); in IsAllReduceMovable()
528 user, accumulation_buffer, in IsAllReduceMovable()
620 HloInstruction* accumulation_buffer = in CreateSinkedAllReduces() local
625 HloInstruction* all_reduce_operand = accumulation_buffer; in CreateSinkedAllReduces()
633 all_reduce_shape, accumulation_buffer)); in CreateSinkedAllReduces()
/external/XNNPACK/src/f32-argmaxpool/
D9p8x-scalar-c1.c18 float* accumulation_buffer, in xnn_f32_argmaxpool_ukernel_9p8x__scalar_c1() argument
32 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__scalar_c1()
133 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__scalar_c1()
239 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__scalar_c1()
D9p8x-neon-c4.c19 float* accumulation_buffer, in xnn_f32_argmaxpool_ukernel_9p8x__neon_c4() argument
33 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__neon_c4()
128 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__neon_c4()
232 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__neon_c4()
D9p8x-wasmsimd-c4.c19 float* accumulation_buffer, in xnn_f32_argmaxpool_ukernel_9p8x__wasmsimd_c4() argument
33 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__wasmsimd_c4()
139 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__wasmsimd_c4()
253 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__wasmsimd_c4()
D9p8x-sse2-c4.c19 float* accumulation_buffer, in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4() argument
33 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4()
139 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4()
253 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4()
/external/XNNPACK/src/xnnpack/
Dargmaxpool.h48 float* accumulation_buffer, \
Dmicrofnptr.h919 void* accumulation_buffer,
932 float* accumulation_buffer,
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.cc204 float* accumulation_buffer = gate; in CalculateLstmGateFloat() local
207 accumulation_buffer, output, n_cell, in CalculateLstmGateFloat()
209 std::swap(accumulation_buffer, output); in CalculateLstmGateFloat()
215 accumulation_buffer, output, n_cell, in CalculateLstmGateFloat()
217 std::swap(accumulation_buffer, output); in CalculateLstmGateFloat()
221 accumulation_buffer, output, n_cell, in CalculateLstmGateFloat()
/external/XNNPACK/src/amalgam/
Dsse2.c430 float* accumulation_buffer, in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4() argument
444 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4()
550 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4()
664 float* ab = accumulation_buffer; in xnn_f32_argmaxpool_ukernel_9p8x__sse2_c4()