Home
last modified time | relevance | path

Searched refs:HandleReduce (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dreduction_dimension_grouper.cc39 Status HandleReduce(HloInstruction *reduce) override { in HandleReduce() function in xla::gpu::ReduceDimensionGroupVisitor
Dreduction_degenerate_dim_remover.cc39 Status HandleReduce(HloInstruction *instr) override { in HandleReduce() function in xla::gpu::ReductionDegenerateDimRemoverVisitor
Dreduction_layout_normalizer.cc38 Status HandleReduce(HloInstruction *reduce) override { in HandleReduce() function in xla::gpu::EnforceMinorToMajorReduceOpVisitor
Dreduction_splitter.cc30 Status HandleReduce(HloInstruction *reduce) override { in HandleReduce() function in xla::gpu::ReductionSplitterVisitor
Dtree_reduction_rewriter.cc58 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::ReductionRewriterVisitor
Dir_emitter_unnested.h190 Status HandleReduce(HloInstruction* reduce) override;
Dir_emitter_unnested.cc2137 Status IrEmitterUnnested::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/
Dtree_reduction_rewriter.cc44 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::ReductionRewriterVisitor
Dhlo_evaluator.h261 Status HandleReduce(HloInstruction* reduce) override;
Dhlo_cost_analysis.h95 Status HandleReduce(const HloInstruction* reduce) override;
Ddfs_hlo_visitor_with_default.h183 Status HandleReduce(HloInstructionPtr reduce) override { in HandleReduce() function
Dhlo_verifier.h77 Status HandleReduce(HloInstruction* reduce) override;
Ddfs_hlo_visitor.h244 virtual Status HandleReduce(HloInstructionPtr hlo) = 0;
Ddynamic_dimension_inference.cc93 Status HandleReduce(HloInstruction* hlo) override;
411 Status DynamicDimensionInferenceVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.cc385 Status HloCostAnalysis::HandleReduce(const HloInstruction* reduce) { in HandleReduce() function in xla::HloCostAnalysis
Dhlo_verifier.cc659 Status ShapeVerifier::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::ShapeVerifier
Dhlo_instruction_test.cc99 Status HandleReduce(HloInstruction* reduce) override { in HandleReduce() function in xla::__anonabde1bd10111::OpAndUserCollectingVisitor
Dhlo_evaluator.cc2404 Status HloEvaluator::HandleReduce(HloInstruction* instr) { in HandleReduce() function in xla::HloEvaluator
Dalgebraic_simplifier.cc343 Status HandleReduce(HloInstruction* hlo) override;
4512 Status AlgebraicSimplifierVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::AlgebraicSimplifierVisitor
Dhlo_instruction.cc3153 return visitor->HandleReduce(this); in Visit()
/external/tensorflow/tensorflow/core/kernels/
Dtile_ops.cc443 HandleReduce<T, NDIM, (D)>(context, reduction_dims, result); \ in HandleCaseImpl()
488 void HandleReduce(OpKernelContext* context, in HandleReduce() function in tensorflow::TileGradientOp
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h148 Status HandleReduce(HloInstruction* reduce) override;
Dir_emitter.cc1794 Status IrEmitter::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h423 Status HandleReduce(HloInstruction* hlo) override;
Dspmd_partitioner.cc2701 Status SpmdPartitioningVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::spmd::SpmdPartitioningVisitor