[MaskR-CNN] Add int8 aabb bbox_transform op
Summary: Adds support for Eigen Utils for custom type defs.
Reviewed By: vkuzo
Differential Revision: D23753697
fbshipit-source-id: de1cfb1c8176a08dd418364f2fce003344fe25bb
diff --git a/caffe2/utils/eigen_utils.h b/caffe2/utils/eigen_utils.h
index 83e7cb2..d5dbe12 100644
--- a/caffe2/utils/eigen_utils.h
+++ b/caffe2/utils/eigen_utils.h
@@ -73,17 +73,28 @@
using EArrXd = Eigen::ArrayXd;
using EArrXi = Eigen::ArrayXi;
using EArrXb = EArrXt<bool>;
+using EArrXI32 = EArrXt<int32_t>;
+using EArrXU16 = EArrXt<uint16_t>;
+using EArrXU8 = EArrXt<uint8_t>;
// 2-d array, column major
template <typename T>
using EArrXXt = Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic>;
using EArrXXf = Eigen::ArrayXXf;
+using EArrXXI32 = EArrXXt<int32_t>;
+using EArrXXU16 = EArrXXt<uint16_t>;
+using EArrXXU8 = EArrXXt<uint8_t>;
+using EArrXXi = EArrXXt<int>;
// 2-d array, row major
template <typename T>
using ERArrXXt =
Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic, Eigen::RowMajor>;
using ERArrXXf = ERArrXXt<float>;
+using ERArrXXI32t = ERArrXXt<int32_t>;
+using ERArrXXU16t = ERArrXXt<uint16_t>;
+using ERArrXXU8t = ERArrXXt<uint8_t>;
+using ERArrXXi = ERArrXXt<int>;
// 1-d vector
template <typename T>
