Searched refs:HloInfeedInstruction (Results 1 – 5 of 5) sorted by relevance
1772 HloInfeedInstruction::HloInfeedInstruction(const Shape& infeed_shape, in HloInfeedInstruction() function in xla::HloInfeedInstruction1782 HloInstructionProto HloInfeedInstruction::ToProto() const { in ToProto()1788 std::vector<string> HloInfeedInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()1796 bool HloInfeedInstruction::IdenticalSlowPath( in IdenticalSlowPath()1804 std::unique_ptr<HloInstruction> HloInfeedInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()1808 return absl::make_unique<HloInfeedInstruction>( in CloneWithNewOperandsImpl()
247 HloInfeedInstruction* infeed = Cast<HloInfeedInstruction>(instruction); in HandleInfeed()
962 class HloInfeedInstruction : public HloInstruction {964 explicit HloInfeedInstruction(const Shape& infeed_shape,
865 return absl::make_unique<HloInfeedInstruction>(infeed_shape, token_operand, in CreateInfeed()3397 return Cast<HloInfeedInstruction>(this)->infeed_config(); in infeed_config()3401 return Cast<HloInfeedInstruction>(this)->set_infeed_config(config); in set_infeed_config()
326 HloInfeedInstruction* infeed = Cast<HloInfeedInstruction>(instruction); in HandleInfeed()