Searched refs:HloBroadcastInstruction (Results 1 – 4 of 4) sorted by relevance
869 HloBroadcastInstruction::HloBroadcastInstruction( in HloBroadcastInstruction() function in xla::HloBroadcastInstruction877 HloInstructionProto HloBroadcastInstruction::ToProto() const { in ToProto()885 std::vector<string> HloBroadcastInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()890 bool HloBroadcastInstruction::IdenticalSlowPath( in IdenticalSlowPath()894 const auto& casted_other = static_cast<const HloBroadcastInstruction&>(other); in IdenticalSlowPath()899 HloBroadcastInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()903 return absl::make_unique<HloBroadcastInstruction>(shape, new_operands[0], in CloneWithNewOperandsImpl()
555 class HloBroadcastInstruction : public HloInstruction {557 explicit HloBroadcastInstruction(const Shape& shape, HloInstruction* operand,
1103 return absl::make_unique<HloBroadcastInstruction>(shape, operand, in CreateBroadcast()
146 CastOrNull<HloBroadcastInstruction>(bias_broadcast_instr); in FindConvWithRelu()