Home
last modified time | relevance | path

Searched refs:HandleSelect (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h95 Status HandleSelect(HloInstruction* select) override;
Dir_emitter_unnested.h177 Status HandleSelect(HloInstruction* select) override;
Dir_emitter.cc418 Status IrEmitter::HandleSelect(HloInstruction* select) { in HandleSelect() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc1205 Status IrEmitterUnnested::HandleSelect(HloInstruction* select) { in HandleSelect() function in xla::gpu::IrEmitterUnnested
1206 return IrEmitter::HandleSelect(select); in HandleSelect()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc60 Status HandleSelect(HloInstruction* hlo) override;
321 Status DynamicDimensionInferenceVisitor::HandleSelect(HloInstruction* hlo) { in HandleSelect() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.h58 Status HandleSelect(const HloInstruction* hlo) override;
Ddfs_hlo_visitor_with_default.h79 Status HandleSelect(HloInstructionPtr select) override { in HandleSelect() function
Dhlo_evaluator.h221 Status HandleSelect(HloInstruction* select) override;
Dhlo_verifier.h45 Status HandleSelect(HloInstruction* select) override;
Ddfs_hlo_visitor.h78 virtual Status HandleSelect(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc179 Status HloCostAnalysis::HandleSelect(const HloInstruction* hlo) { in HandleSelect() function in xla::HloCostAnalysis
Dhlo_verifier.cc112 Status ShapeVerifier::HandleSelect(HloInstruction* select) { in HandleSelect() function in xla::ShapeVerifier
Dhlo_evaluator.cc1288 Status HloEvaluator::HandleSelect(HloInstruction* select) { in HandleSelect() function in xla::HloEvaluator
Dalgebraic_simplifier.cc238 Status HandleSelect(HloInstruction* select) override;
3308 Status AlgebraicSimplifierVisitor::HandleSelect(HloInstruction* select) { in HandleSelect() function in xla::AlgebraicSimplifierVisitor
Dhlo_evaluator_typed_visitor.h964 Status HandleSelect(HloInstruction* select) override { in HandleSelect() function
Dhlo_instruction.cc2542 return visitor->HandleSelect(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h152 Status HandleSelect(HloInstruction* select) override;
Dir_emitter.cc305 Status IrEmitter::HandleSelect(HloInstruction* select) { in HandleSelect() function in xla::cpu::IrEmitter