/external/tensorflow/tensorflow/compiler/xla/service/ |
D | reduce_decomposer.cc | 36 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
|
D | tree_reduction_rewriter.cc | 45 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::ReductionRewriterVisitor
|
D | hlo_cost_analysis.h | 139 Status HandleReduce(const HloInstruction* reduce) override;
|
D | dfs_hlo_visitor_with_default.h | 196 Status HandleReduce(HloInstructionPtr reduce) override { in HandleReduce() function
|
D | hlo_evaluator.h | 337 Status HandleReduce(HloInstruction* reduce) override;
|
D | algebraic_simplifier.h | 319 Status HandleReduce(HloInstruction* hlo) override;
|
D | dfs_hlo_visitor.h | 250 virtual Status HandleReduce(HloInstructionPtr hlo) = 0;
|
D | hlo_verifier.h | 178 Status HandleReduce(HloInstruction* reduce) override;
|
D | dynamic_dimension_inference.cc | 107 Status HandleReduce(HloInstruction* hlo) override; 444 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_instruction_test.cc | 100 Status HandleReduce(HloInstruction* reduce) override { in HandleReduce() function in xla::__anon4922aaaf0111::OpAndUserCollectingVisitor
|
D | hlo_verifier.cc | 1022 Status ShapeVerifier::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::ShapeVerifier
|
D | hlo_evaluator.cc | 3709 Status HloEvaluator::HandleReduce(HloInstruction* instr) { in HandleReduce() function in xla::HloEvaluator
|
D | hlo_instruction.cc | 3580 return visitor->HandleReduce(this); in Visit()
|
D | algebraic_simplifier.cc | 5279 Status AlgebraicSimplifierVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::AlgebraicSimplifierVisitor
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | reduction_dimension_grouper.cc | 36 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::ReduceDimensionGroupVisitor
|
D | reduction_splitter.cc | 30 Status HandleReduce(HloInstruction *reduce) override { in HandleReduce() function in xla::gpu::ReductionSplitterVisitor
|
D | reduction_degenerate_dim_remover.cc | 39 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::ReductionDegenerateDimRemoverVisitor
|
D | reduction_layout_normalizer.cc | 38 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::EnforceMinorToMajorReduceOpVisitor
|
D | tree_reduction_rewriter.cc | 52 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::ReductionRewriterVisitor
|
/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 | 156 Status HandleReduce(HloInstruction* reduce) override;
|
D | ir_emitter.cc | 1900 Status IrEmitter::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::cpu::IrEmitter
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | spmd_partitioner.h | 486 Status HandleReduce(HloInstruction* hlo) override;
|
D | spmd_partitioner.cc | 3470 Status SpmdPartitioningVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::spmd::SpmdPartitioningVisitor
|