Searched refs:HloRngInstruction (Results 1 – 3 of 3) sorted by relevance
1599 HloRngInstruction::HloRngInstruction( in HloRngInstruction() function in xla::HloRngInstruction1608 HloInstructionProto HloRngInstruction::ToProto() const { in ToProto()1614 std::vector<string> HloRngInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()1619 bool HloRngInstruction::IsElementwiseImpl( in IsElementwiseImpl()1624 bool HloRngInstruction::IdenticalSlowPath( in IdenticalSlowPath()1631 std::unique_ptr<HloInstruction> HloRngInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()1634 return absl::make_unique<HloRngInstruction>(shape, distribution_, in CloneWithNewOperandsImpl()
834 class HloRngInstruction : public HloInstruction {836 explicit HloRngInstruction(const Shape& shape,
668 return absl::make_unique<HloRngInstruction>(shape, distribution, parameters); in CreateRng()3364 return Cast<HloRngInstruction>(this)->random_distribution(); in random_distribution()