Searched refs:HloOutfeedInstruction (Results 1 – 4 of 4) sorted by relevance
1812 HloOutfeedInstruction::HloOutfeedInstruction(const Shape& outfeed_shape, in HloOutfeedInstruction() function in xla::HloOutfeedInstruction1823 HloInstructionProto HloOutfeedInstruction::ToProto() const { in ToProto()1830 std::vector<string> HloOutfeedInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()1838 bool HloOutfeedInstruction::IdenticalSlowPath( in IdenticalSlowPath()1846 std::unique_ptr<HloInstruction> HloOutfeedInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()1850 return absl::make_unique<HloOutfeedInstruction>( in CloneWithNewOperandsImpl()
257 HloOutfeedInstruction* outfeed = Cast<HloOutfeedInstruction>(instruction); in HandleOutfeed()
998 class HloOutfeedInstruction : public HloInstruction {1000 explicit HloOutfeedInstruction(const Shape& outfeed_shape,
872 return absl::make_unique<HloOutfeedInstruction>( in CreateOutfeed()3405 return Cast<HloOutfeedInstruction>(this)->outfeed_shape(); in outfeed_shape()3409 return Cast<HloOutfeedInstruction>(this)->outfeed_config(); in outfeed_config()