Searched refs:CreateRngGetAndUpdateState (Results 1 – 4 of 4) sorted by relevance
159 computation->AddInstruction(HloInstruction::CreateRngGetAndUpdateState( in ExpandInstruction()
545 static std::unique_ptr<HloInstruction> CreateRngGetAndUpdateState(
377 instruction = CreateRngGetAndUpdateState(shape, proto.delta()); in CreateFromProto()858 HloInstruction::CreateRngGetAndUpdateState(const Shape& shape, int64 delta) { in CreateRngGetAndUpdateState() function in xla::HloInstruction
2142 HloInstruction::CreateRngGetAndUpdateState(shape, *delta)); in ParseInstructionRhs()