Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h72 Status HandleReduce(HloInstruction* reduce) override;
Dir_emitter.h88 Status HandleReduce(HloInstruction* reduce) override;
Dir_emitter.cc627 Status IrEmitter::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc1660 Status IrEmitterUnnested::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::gpu::IrEmitterUnnested
1693 return IrEmitter::HandleReduce(reduce); in HandleReduce()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.h54 Status HandleReduce(HloInstruction* reduce) override;
Dhlo_cost_analysis.h79 Status HandleReduce(const HloInstruction* reduce) override;
Ddfs_hlo_visitor_with_default.h149 Status HandleReduce(HloInstructionPtr reduce) override { in HandleReduce() function
Ddfs_hlo_visitor.h199 virtual Status HandleReduce(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc251 Status HloCostAnalysis::HandleReduce(const HloInstruction* reduce) { in HandleReduce() function in xla::HloCostAnalysis
Dhlo_verifier.cc157 Status ShapeVerifier::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::ShapeVerifier
Dalgebraic_simplifier.cc159 Status HandleReduce(HloInstruction* reduce) override;
1610 Status AlgebraicSimplifierVisitor::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::AlgebraicSimplifierVisitor
Dhlo_instruction_test.cc95 Status HandleReduce(HloInstruction* reduce) override { in HandleReduce() function in xla::__anon971548e70111::OpAndUserCollectingVisitor
Dhlo_evaluator.cc1435 Status HandleReduce(HloInstruction* reduce) override { in HandleReduce() function in xla::HloEvaluator::TypedVisitor
Dhlo_instruction.cc2599 return visitor->HandleReduce(this); in Visit()
/external/tensorflow/tensorflow/core/kernels/
Dtile_ops.cc369 HandleReduce<T, NDIM, (D)>(context, reduction_dims, result); \ in HandleCaseImpl()
414 void HandleReduce(OpKernelContext* context, in HandleReduce() function in tensorflow::TileGradientOp
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h133 Status HandleReduce(HloInstruction* reduce) override;
Dir_emitter.cc1737 Status IrEmitter::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::cpu::IrEmitter