Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_spmd_partitioner.cc26 Status GpuSpmdPartitioningVisitor::HandleRngGetAndUpdateState( in HandleRngGetAndUpdateState() function in xla::gpu::GpuSpmdPartitioningVisitor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dstateful_rng_spmd_partitioner.cc26 Status StatefulRngSpmdPartitioningVisitor::HandleRngGetAndUpdateState( in HandleRngGetAndUpdateState() function in xla::spmd::StatefulRngSpmdPartitioningVisitor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddfs_hlo_visitor_with_default.h144 Status HandleRngGetAndUpdateState(HloInstructionPtr random) override { in HandleRngGetAndUpdateState() function
H A Dhlo_cost_analysis.cc864 Status HloCostAnalysis::HandleRngGetAndUpdateState( in HandleRngGetAndUpdateState() function in xla::HloCostAnalysis
H A Dhlo_verifier.cc876 Status ShapeVerifier::HandleRngGetAndUpdateState(HloInstruction* instruction) { in HandleRngGetAndUpdateState() function in xla::ShapeVerifier
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc2939 Status IrEmitter::HandleRngGetAndUpdateState(HloInstruction* rng_state) { in HandleRngGetAndUpdateState() function in xla::cpu::IrEmitter