Searched refs:HloRngGetAndUpdateStateInstruction (Results 1 – 5 of 5) sorted by relevance
3154 HloRngGetAndUpdateStateInstruction::HloRngGetAndUpdateStateInstruction( in HloRngGetAndUpdateStateInstruction() function in xla::HloRngGetAndUpdateStateInstruction3158 HloInstructionProto HloRngGetAndUpdateStateInstruction::ToProto() const { in ToProto()3165 HloRngGetAndUpdateStateInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()3170 bool HloRngGetAndUpdateStateInstruction::IdenticalSlowPath( in IdenticalSlowPath()3175 static_cast<const HloRngGetAndUpdateStateInstruction&>(other); in IdenticalSlowPath()3180 HloRngGetAndUpdateStateInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()3186 return absl::make_unique<HloRngGetAndUpdateStateInstruction>(shape, delta()); in CloneWithNewOperandsImpl()
1904 class HloRngGetAndUpdateStateInstruction : public HloInstruction {1906 explicit HloRngGetAndUpdateStateInstruction(const Shape& shape, int64 delta);
859 return absl::make_unique<HloRngGetAndUpdateStateInstruction>(shape, delta); in CreateRngGetAndUpdateState()
1279 auto hlo_rng = xla::Cast<xla::HloRngGetAndUpdateStateInstruction>(instr); in EmitRngGetAndUpdateStateOp()
2761 Cast<HloRngGetAndUpdateStateInstruction>(rng_state)->delta(), module_, in HandleRngGetAndUpdateState()