Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dtile_ops.cc421 void HandleReduce(OpKernelContext* context, in HandleReduce() function in tensorflow::TileGradientOp
/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default.h164 Status HandleReduce(HloInstructionPtr reduce) override { in HandleReduce() function
Ddynamic_dimension_inference.cc166 Status DynamicDimensionInferenceVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::DynamicDimensionInferenceVisitor
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
Dhlo_evaluator_typed_visitor.h1684 Status HandleReduce(HloInstruction* hlo) override { in HandleReduce() function
Dhlo_instruction_test.cc97 Status HandleReduce(HloInstruction* reduce) override { in HandleReduce() function in xla::__anon9b186d4a0111::OpAndUserCollectingVisitor
Dalgebraic_simplifier.cc2908 Status AlgebraicSimplifierVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::AlgebraicSimplifierVisitor
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
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
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc1872 Status IrEmitter::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::cpu::IrEmitter