Home
last modified time | relevance | path

Searched refs:HandleRngGetAndUpdateState (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h91 Status HandleRngGetAndUpdateState(const HloInstruction* random) override;
Ddfs_hlo_visitor_with_default.h131 Status HandleRngGetAndUpdateState(HloInstructionPtr random) override { in HandleRngGetAndUpdateState() function
Dhlo_verifier.h72 Status HandleRngGetAndUpdateState(HloInstruction*) override;
Ddfs_hlo_visitor.h238 virtual Status HandleRngGetAndUpdateState(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc809 Status HloCostAnalysis::HandleRngGetAndUpdateState( in HandleRngGetAndUpdateState() function in xla::HloCostAnalysis
Dhlo_verifier.cc523 Status ShapeVerifier::HandleRngGetAndUpdateState(HloInstruction* instruction) { in HandleRngGetAndUpdateState() function in xla::ShapeVerifier
Dhlo_instruction.cc3229 return visitor->HandleRngGetAndUpdateState(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h172 Status HandleRngGetAndUpdateState(HloInstruction* rng_state) override;
Dir_emitter.cc2758 Status IrEmitter::HandleRngGetAndUpdateState(HloInstruction* rng_state) { in HandleRngGetAndUpdateState() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h199 Status HandleRngGetAndUpdateState(HloInstruction* rng_state) override;
Dir_emitter_unnested.cc2454 Status IrEmitterUnnested::HandleRngGetAndUpdateState( in HandleRngGetAndUpdateState() function in xla::gpu::IrEmitterUnnested