Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dreduce_decomposer.cc36 Status HandleReduce(HloInstruction* hlo) override { in HandleReduce() function in xla::__anona69c36340111::VariadicReductionLayoutEqualizer
73 Status HandleReduce(HloInstruction* hlo) override { in HandleReduce() function in xla::__anona69c36340111::ReduceDecomposerVisitor
Dtree_reduction_rewriter.cc45 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::ReductionRewriterVisitor
Dhlo_cost_analysis.h139 Status HandleReduce(const HloInstruction* reduce) override;
Ddfs_hlo_visitor_with_default.h196 Status HandleReduce(HloInstructionPtr reduce) override { in HandleReduce() function
Dhlo_evaluator.h337 Status HandleReduce(HloInstruction* reduce) override;
Dalgebraic_simplifier.h319 Status HandleReduce(HloInstruction* hlo) override;
Ddfs_hlo_visitor.h250 virtual Status HandleReduce(HloInstructionPtr hlo) = 0;
Dhlo_verifier.h178 Status HandleReduce(HloInstruction* reduce) override;
Ddynamic_dimension_inference.cc107 Status HandleReduce(HloInstruction* hlo) override;
444 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_instruction_test.cc100 Status HandleReduce(HloInstruction* reduce) override { in HandleReduce() function in xla::__anon4922aaaf0111::OpAndUserCollectingVisitor
Dhlo_verifier.cc1022 Status ShapeVerifier::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::ShapeVerifier
Dhlo_evaluator.cc3709 Status HloEvaluator::HandleReduce(HloInstruction* instr) { in HandleReduce() function in xla::HloEvaluator
Dhlo_instruction.cc3580 return visitor->HandleReduce(this); in Visit()
Dalgebraic_simplifier.cc5279 Status AlgebraicSimplifierVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::AlgebraicSimplifierVisitor
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dreduction_dimension_grouper.cc36 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::ReduceDimensionGroupVisitor
Dreduction_splitter.cc30 Status HandleReduce(HloInstruction *reduce) override { in HandleReduce() function in xla::gpu::ReductionSplitterVisitor
Dreduction_degenerate_dim_remover.cc39 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::ReductionDegenerateDimRemoverVisitor
Dreduction_layout_normalizer.cc38 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::EnforceMinorToMajorReduceOpVisitor
Dtree_reduction_rewriter.cc52 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::ReductionRewriterVisitor
/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.h156 Status HandleReduce(HloInstruction* reduce) override;
Dir_emitter.cc1900 Status IrEmitter::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h486 Status HandleReduce(HloInstruction* hlo) override;
Dspmd_partitioner.cc3470 Status SpmdPartitioningVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::spmd::SpmdPartitioningVisitor