Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h89 Status HandleRng(const HloInstruction* random) override;
Ddfs_hlo_visitor_with_default.h125 Status HandleRng(HloInstructionPtr random) override { in HandleRng() function
Dhlo_verifier.h70 Status HandleRng(HloInstruction*) override;
Ddfs_hlo_visitor.h236 virtual Status HandleRng(HloInstructionPtr hlo) = 0;
Dhlo_evaluator_typed_visitor.h2709 Status HandleRng(HloInstruction* random) {
2715 Status HandleRng(HloInstruction* random) {
2775 Status HandleRng(HloInstruction* random) {
2811 Status HandleRng(HloInstruction* random) override {
2812 return HandleRng<ReturnT>(random);
Dhlo_cost_analysis.cc791 Status HloCostAnalysis::HandleRng(const HloInstruction* random) { in HandleRng() function in xla::HloCostAnalysis
Dhlo_verifier.cc457 Status ShapeVerifier::HandleRng(HloInstruction* instruction) { in HandleRng() function in xla::ShapeVerifier
Dhlo_instruction.cc3225 return visitor->HandleRng(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h171 Status HandleRng(HloInstruction* rng) override;
Dir_emitter.cc2754 Status IrEmitter::HandleRng(HloInstruction* rng) { in HandleRng() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h430 Status HandleRng(HloInstruction* hlo) override;
Dspmd_partitioner.cc3043 Status SpmdPartitioningVisitor::HandleRng(HloInstruction* hlo) { in HandleRng() function in xla::spmd::SpmdPartitioningVisitor
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h198 Status HandleRng(HloInstruction* random) override;
Dir_emitter_unnested.cc2450 Status IrEmitterUnnested::HandleRng(HloInstruction* rng) { in HandleRng() function in xla::gpu::IrEmitterUnnested