Searched refs:HloTransposeInstruction (Results 1 – 3 of 3) sorted by relevance
824 HloTransposeInstruction::HloTransposeInstruction( in HloTransposeInstruction() function in xla::HloTransposeInstruction832 bool HloTransposeInstruction::IsRank2Transpose() const { in IsRank2Transpose()839 HloInstructionProto HloTransposeInstruction::ToProto() const { in ToProto()847 std::vector<string> HloTransposeInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()852 bool HloTransposeInstruction::IdenticalSlowPath( in IdenticalSlowPath()856 const auto& casted_other = static_cast<const HloTransposeInstruction&>(other); in IdenticalSlowPath()861 HloTransposeInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()865 return absl::make_unique<HloTransposeInstruction>(shape, new_operands[0], in CloneWithNewOperandsImpl()
528 class HloTransposeInstruction : public HloInstruction {530 explicit HloTransposeInstruction(const Shape& shape, HloInstruction* operand,
1189 return absl::make_unique<HloTransposeInstruction>(shape, operand, dimensions); in CreateTranspose()3243 auto transpose = DynCast<HloTransposeInstruction>(this); in IsRank2Transpose()