Home
last modified time | relevance | path

Searched defs:reduce_window (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtree_reduction_rewriter.cc83 HloInstruction *reduce_window = in HandleReduce() local
Dconvolution_group_converter.cc397 auto reduce_window = add(HloInstruction::CreateReduceWindow( in HandleBatchGroupCount() local
Ddfs_hlo_visitor_with_default.h186 Status HandleReduceWindow(HloInstructionPtr reduce_window) override { in HandleReduceWindow()
Dhlo_cost_analysis.cc410 const HloInstruction* reduce_window) { in HandleReduceWindow()
Ddynamic_dimension_inference_test.cc971 auto* reduce_window = in TEST_F() local
Dhlo_verifier.cc892 Status ShapeVerifier::HandleReduceWindow(HloInstruction* reduce_window) { in HandleReduceWindow()
Ddynamic_dimension_inference.cc1063 HloInstruction* reduce_window = hlo; in HandleReduceWindow() local
Dsharding_propagation_test.cc1376 auto* reduce_window = FindInstruction(module.get(), "reduce-window"); in TEST_P() local
Delemental_ir_emitter.cc2590 const HloReduceWindowInstruction* reduce_window, in EmitElementalReduceWindow()
Dalgebraic_simplifier.cc4764 HloInstruction* reduce_window) { in HandleReduceWindow()
Dalgebraic_simplifier_test.cc4387 HloInstruction* reduce_window = in TEST_F() local
4473 HloInstruction* reduce_window = in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py348 def reduce_window(operand, function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc647 Status IrEmitter::HandleReduceWindow(HloInstruction* reduce_window) { in HandleReduceWindow()