Searched refs:HloCopyStartInstruction (Results 1 – 3 of 3) sorted by relevance
208 HloCopyStartInstruction::HloCopyStartInstruction(const Shape& shape, in HloCopyStartInstruction() function in xla::HloCopyStartInstruction216 HloInstructionProto HloCopyStartInstruction::ToProto() const { in ToProto()222 std::vector<string> HloCopyStartInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()231 bool HloCopyStartInstruction::IdenticalSlowPath( in IdenticalSlowPath()235 const auto& casted_other = static_cast<const HloCopyStartInstruction&>(other); in IdenticalSlowPath()241 HloCopyStartInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()245 return absl::make_unique<HloCopyStartInstruction>( in CloneWithNewOperandsImpl()
136 class HloCopyStartInstruction : public HloInstruction {138 explicit HloCopyStartInstruction(const Shape& shape, HloInstruction* operand,
1003 return absl::make_unique<HloCopyStartInstruction>(shape, operand, in CreateCopyStart()4262 return Cast<HloCopyStartInstruction>(this)->is_cross_program_prefetch(); in is_cross_program_prefetch()