Searched refs:HloReverseInstruction (Results 1 – 3 of 3) sorted by relevance
651 HloReverseInstruction::HloReverseInstruction(const Shape& shape, in HloReverseInstruction() function in xla::HloReverseInstruction659 HloInstructionProto HloReverseInstruction::ToProto() const { in ToProto()667 std::vector<string> HloReverseInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()672 bool HloReverseInstruction::IdenticalSlowPath( in IdenticalSlowPath()676 const auto& casted_other = static_cast<const HloReverseInstruction&>(other); in IdenticalSlowPath()680 std::unique_ptr<HloInstruction> HloReverseInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()684 return absl::make_unique<HloReverseInstruction>(shape, new_operands[0], in CloneWithNewOperandsImpl()
399 class HloReverseInstruction : public HloInstruction {401 explicit HloReverseInstruction(const Shape& shape, HloInstruction* operand,
911 return absl::make_unique<HloReverseInstruction>(shape, operand, dimensions); in CreateReverse()