/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | reduction_dimension_grouper.cc | 39 Status HandleReduce(HloInstruction *reduce) override { in HandleReduce() function in xla::gpu::ReduceDimensionGroupVisitor
|
D | reduction_degenerate_dim_remover.cc | 39 Status HandleReduce(HloInstruction *instr) override { in HandleReduce() function in xla::gpu::ReductionDegenerateDimRemoverVisitor
|
D | reduction_layout_normalizer.cc | 38 Status HandleReduce(HloInstruction *reduce) override { in HandleReduce() function in xla::gpu::EnforceMinorToMajorReduceOpVisitor
|
D | reduction_splitter.cc | 30 Status HandleReduce(HloInstruction *reduce) override { in HandleReduce() function in xla::gpu::ReductionSplitterVisitor
|
D | tree_reduction_rewriter.cc | 58 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::ReductionRewriterVisitor
|
D | ir_emitter_unnested.h | 190 Status HandleReduce(HloInstruction* reduce) override;
|
D | ir_emitter_unnested.cc | 2137 Status IrEmitterUnnested::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::gpu::IrEmitterUnnested
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | tree_reduction_rewriter.cc | 44 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::ReductionRewriterVisitor
|
D | hlo_evaluator.h | 261 Status HandleReduce(HloInstruction* reduce) override;
|
D | hlo_cost_analysis.h | 95 Status HandleReduce(const HloInstruction* reduce) override;
|
D | dfs_hlo_visitor_with_default.h | 183 Status HandleReduce(HloInstructionPtr reduce) override { in HandleReduce() function
|
D | hlo_verifier.h | 77 Status HandleReduce(HloInstruction* reduce) override;
|
D | dfs_hlo_visitor.h | 244 virtual Status HandleReduce(HloInstructionPtr hlo) = 0;
|
D | dynamic_dimension_inference.cc | 93 Status HandleReduce(HloInstruction* hlo) override; 411 Status DynamicDimensionInferenceVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::DynamicDimensionInferenceVisitor
|
D | hlo_cost_analysis.cc | 385 Status HloCostAnalysis::HandleReduce(const HloInstruction* reduce) { in HandleReduce() function in xla::HloCostAnalysis
|
D | hlo_verifier.cc | 659 Status ShapeVerifier::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::ShapeVerifier
|
D | hlo_instruction_test.cc | 99 Status HandleReduce(HloInstruction* reduce) override { in HandleReduce() function in xla::__anonabde1bd10111::OpAndUserCollectingVisitor
|
D | hlo_evaluator.cc | 2404 Status HloEvaluator::HandleReduce(HloInstruction* instr) { in HandleReduce() function in xla::HloEvaluator
|
D | algebraic_simplifier.cc | 343 Status HandleReduce(HloInstruction* hlo) override; 4512 Status AlgebraicSimplifierVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::AlgebraicSimplifierVisitor
|
D | hlo_instruction.cc | 3153 return visitor->HandleReduce(this); in Visit()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | tile_ops.cc | 443 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/ |
D | ir_emitter.h | 148 Status HandleReduce(HloInstruction* reduce) override;
|
D | ir_emitter.cc | 1794 Status IrEmitter::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::cpu::IrEmitter
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | spmd_partitioner.h | 423 Status HandleReduce(HloInstruction* hlo) override;
|
D | spmd_partitioner.cc | 2701 Status SpmdPartitioningVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::spmd::SpmdPartitioningVisitor
|