Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dapprox_topk.cc125 auto reduction_computation = in AggregateToTopKBuilder() local
182 auto reduction_computation = in ApproxTopK() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dreduction_ops_common.cc124 xla::XlaComputation reduction_computation = r.Build().value(); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_hlo_schedule_test.cc184 HloComputation* reduction_computation = in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dwhile_loop_all_reduce_code_motion_test.cc126 HloComputation* reduction_computation = in TEST_F() local