Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h92 Status HandleReduce(HloInstruction* reduce) override;
Dir_emitter_unnested.h169 Status HandleReduce(HloInstruction* reduce) override;
Dir_emitter.cc664 Status IrEmitter::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc722 Status IrEmitterUnnested::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::gpu::IrEmitterUnnested
731 return IrEmitter::HandleReduce(reduce); in HandleReduce()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc44 Status HandleReduce(HloInstruction* hlo) override;
166 Status DynamicDimensionInferenceVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.h86 Status HandleReduce(const HloInstruction* reduce) override;
Ddfs_hlo_visitor_with_default.h164 Status HandleReduce(HloInstructionPtr reduce) override { in HandleReduce() function
Dhlo_evaluator.h239 Status HandleReduce(HloInstruction* reduce) override;
Dhlo_verifier.h69 Status HandleReduce(HloInstruction* reduce) override;
Ddfs_hlo_visitor.h228 virtual Status HandleReduce(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc278 Status HloCostAnalysis::HandleReduce(const HloInstruction* reduce) { in HandleReduce() function in xla::HloCostAnalysis
Dhlo_verifier.cc441 Status ShapeVerifier::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::ShapeVerifier
Dhlo_evaluator.cc1560 Status HloEvaluator::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::HloEvaluator
Dalgebraic_simplifier.cc228 Status HandleReduce(HloInstruction* reduce) override;
2908 Status AlgebraicSimplifierVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::AlgebraicSimplifierVisitor
Dhlo_instruction_test.cc97 Status HandleReduce(HloInstruction* reduce) override { in HandleReduce() function in xla::__anon0e998cf30111::OpAndUserCollectingVisitor
Dhlo_evaluator_typed_visitor.h1684 Status HandleReduce(HloInstruction* hlo) override { in HandleReduce() function
Dhlo_instruction.cc2564 return visitor->HandleReduce(this); in Visit()
/external/tensorflow/tensorflow/core/kernels/
Dtile_ops.cc376 HandleReduce<T, NDIM, (D)>(context, reduction_dims, result); \ in HandleCaseImpl()
421 void HandleReduce(OpKernelContext* context, in HandleReduce() function in tensorflow::TileGradientOp
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h162 Status HandleReduce(HloInstruction* reduce) override;
Dir_emitter.cc1872 Status IrEmitter::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::cpu::IrEmitter