| /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_splitter.cc | 30 Status HandleReduce(HloInstruction *reduce) override { in HandleReduce() function in xla::gpu::ReductionSplitterVisitor
|
| D | reduction_layout_normalizer.cc | 38 Status HandleReduce(HloInstruction *reduce) override { in HandleReduce() function in xla::gpu::EnforceMinorToMajorReduceOpVisitor
|
| D | tree_reduction_rewriter.cc | 58 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::ReductionRewriterVisitor
|
| 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 | dfs_hlo_visitor_with_default.h | 183 Status HandleReduce(HloInstructionPtr reduce) override { in HandleReduce() function
|
| 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 | dynamic_dimension_inference.cc | 411 Status DynamicDimensionInferenceVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::DynamicDimensionInferenceVisitor
|
| 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 | 4512 Status AlgebraicSimplifierVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::AlgebraicSimplifierVisitor
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tile_ops.cc | 488 void HandleReduce(OpKernelContext* context, in HandleReduce() function in tensorflow::TileGradientOp
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| 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.cc | 2701 Status SpmdPartitioningVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::spmd::SpmdPartitioningVisitor
|