Searched refs:HloRngInstruction (Results 1 – 3 of 3) sorted by relevance
1865 HloRngInstruction::HloRngInstruction( in HloRngInstruction() function in xla::HloRngInstruction1874 HloInstructionProto HloRngInstruction::ToProto() const { in ToProto()1880 std::vector<string> HloRngInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()1885 bool HloRngInstruction::IsElementwiseImpl( in IsElementwiseImpl()1890 bool HloRngInstruction::IdenticalSlowPath( in IdenticalSlowPath()1894 const auto& casted_other = static_cast<const HloRngInstruction&>(other); in IdenticalSlowPath()1898 std::unique_ptr<HloInstruction> HloRngInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()1901 return absl::make_unique<HloRngInstruction>(shape, distribution_, in CloneWithNewOperandsImpl()
1015 class HloRngInstruction : public HloInstruction {1017 explicit HloRngInstruction(const Shape& shape,
854 return absl::make_unique<HloRngInstruction>(shape, distribution, parameters); in CreateRng()4059 return Cast<HloRngInstruction>(this)->random_distribution(); in random_distribution()