Home
last modified time | relevance | path

Searched refs:HandleRng (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h82 Status HandleRng(const HloInstruction* random) override;
Ddfs_hlo_visitor_with_default.h112 Status HandleRng(HloInstructionPtr random) override { in HandleRng() function
Dhlo_verifier.h64 Status HandleRng(HloInstruction*) override;
Ddfs_hlo_visitor.h222 virtual Status HandleRng(HloInstructionPtr hlo) = 0;
Dhlo_evaluator_typed_visitor.h2693 Status HandleRng(HloInstruction* random) { in HandleRng() function
2699 Status HandleRng(HloInstruction* random) { in HandleRng() function
2759 Status HandleRng(HloInstruction* random) { in HandleRng() function
2795 Status HandleRng(HloInstruction* random) override { in HandleRng() function
2796 return HandleRng<ReturnT>(random); in HandleRng()
Dhlo_cost_analysis.cc605 Status HloCostAnalysis::HandleRng(const HloInstruction* random) { in HandleRng() function in xla::HloCostAnalysis
Dhlo_verifier.cc282 Status ShapeVerifier::HandleRng(HloInstruction* instruction) { in HandleRng() function in xla::ShapeVerifier
Dhlo_instruction.cc2630 return visitor->HandleRng(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h175 Status HandleRng(HloInstruction* random) override;
Dir_emitter_unnested.cc993 Status IrEmitterUnnested::HandleRng(HloInstruction* rng) { in HandleRng() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h184 Status HandleRng(HloInstruction* rng) override;
Dir_emitter.cc2652 Status IrEmitter::HandleRng(HloInstruction* rng) { in HandleRng() function in xla::cpu::IrEmitter