Home
last modified time | relevance | path

Searched defs:HandleRngGetAndUpdateState (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_spmd_partitioner.cc26 Status GpuSpmdPartitioningVisitor::HandleRngGetAndUpdateState( in HandleRngGetAndUpdateState() function in xla::gpu::GpuSpmdPartitioningVisitor
/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default.h146 Status HandleRngGetAndUpdateState(HloInstructionPtr random) override { in HandleRngGetAndUpdateState() function
Dhlo_cost_analysis.cc838 Status HloCostAnalysis::HandleRngGetAndUpdateState( in HandleRngGetAndUpdateState() function in xla::HloCostAnalysis
Dhlo_verifier.cc860 Status ShapeVerifier::HandleRngGetAndUpdateState(HloInstruction* instruction) { in HandleRngGetAndUpdateState() function in xla::ShapeVerifier
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2795 Status IrEmitter::HandleRngGetAndUpdateState(HloInstruction* rng_state) { in HandleRngGetAndUpdateState() function in xla::cpu::IrEmitter