Home
last modified time | relevance | path

Searched defs:HandleReduce (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dreduction_degenerate_dim_remover.cc39 Status HandleReduce(HloInstruction *instr) override { in HandleReduce() function in xla::gpu::ReductionDegenerateDimRemoverVisitor
Dreduction_dimension_grouper.cc39 Status HandleReduce(HloInstruction *reduce) override { in HandleReduce() function in xla::gpu::ReduceDimensionGroupVisitor
Dreduction_layout_normalizer.cc38 Status HandleReduce(HloInstruction *reduce) override { in HandleReduce() function in xla::gpu::EnforceMinorToMajorReduceOpVisitor
Dtree_reduction_rewriter.cc45 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::ReductionRewriterVisitor
Dir_emitter.cc701 Status IrEmitter::HandleReduce(HloInstruction* instr) { in HandleReduce() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc546 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
Ddfs_hlo_visitor_with_default.h171 Status HandleReduce(HloInstructionPtr reduce) override { in HandleReduce() function
Dhlo_cost_analysis.cc358 Status HloCostAnalysis::HandleReduce(const HloInstruction* reduce) { in HandleReduce() function in xla::HloCostAnalysis
Ddynamic_dimension_inference.cc254 Status DynamicDimensionInferenceVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::DynamicDimensionInferenceVisitor
Dhlo_verifier.cc554 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::__anonee24a0f50111::OpAndUserCollectingVisitor
Dalgebraic_simplifier.cc3550 Status AlgebraicSimplifierVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::AlgebraicSimplifierVisitor
Dhlo_evaluator.cc2384 Status HloEvaluator::HandleReduce(HloInstruction* instr) { in HandleReduce() function in xla::HloEvaluator
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dhlo_dialect_emitter.cc154 Status HloDialectEmitter::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::mlir_gpu::HloDialectEmitter
Dlhlo_dialect_emitter.cc245 Status LhloDialectEmitter::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::mlir_gpu::LhloDialectEmitter
/external/tensorflow/tensorflow/core/kernels/
Dtile_ops.cc519 void HandleReduce(OpKernelContext* context, in HandleReduce() function in tensorflow::TileGradientOp
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2119 Status IrEmitter::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::cpu::IrEmitter