Home
last modified time | relevance | path

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

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