Searched refs:HloParameterInstruction (Results 1 – 5 of 5) sorted by relevance
1905 HloParameterInstruction::HloParameterInstruction(int64 parameter_number, in HloParameterInstruction() function in xla::HloParameterInstruction1913 HloInstructionProto HloParameterInstruction::ToProto() const { in ToProto()1925 std::vector<string> HloParameterInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()1942 string HloParameterInstruction::OperandsToStringWithCanonicalNameMap( in OperandsToStringWithCanonicalNameMap()1948 bool HloParameterInstruction::IdenticalSlowPath( in IdenticalSlowPath()1952 const auto& casted_other = static_cast<const HloParameterInstruction&>(other); in IdenticalSlowPath()1957 HloParameterInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()1960 auto clone = absl::make_unique<HloParameterInstruction>(parameter_number_, in CloneWithNewOperandsImpl()
127 Cast<HloParameterInstruction>(instruction)->parameter_number(); in IsAllReduceMovable()
1100 HloParameterInstruction* loop_body_param) { in TryMergeInductionVariables()1169 Cast<HloParameterInstruction>( in TryMergeInductionVariables()
825 return absl::make_unique<HloParameterInstruction>(parameter_number, shape, in CreateParameter()3478 if (auto hlo_param = DynCast<HloParameterInstruction>(&hlo)) { in ComputeInternal()4063 return Cast<HloParameterInstruction>(this)->parameter_number(); in parameter_number()4068 return Cast<HloParameterInstruction>(this) in set_parameter_replicated_at_leaf_buffers()4075 return Cast<HloParameterInstruction>(this) in set_parameter_replicated_at_leaf_buffers()4082 return Cast<HloParameterInstruction>(this) in parameter_replicated_at_leaf_buffers()
1043 class HloParameterInstruction : public HloInstruction {1045 explicit HloParameterInstruction(int64 parameter_number, const Shape& shape,