diff --git a/mediapipe/tasks/cc/vision/custom_ops/fused_batch_norm.cc b/mediapipe/tasks/cc/vision/custom_ops/fused_batch_norm.cc index b3eccd340..650f723e2 100644 --- a/mediapipe/tasks/cc/vision/custom_ops/fused_batch_norm.cc +++ b/mediapipe/tasks/cc/vision/custom_ops/fused_batch_norm.cc @@ -44,14 +44,11 @@ constexpr int kOutputSavedVarIndex = 4; template struct TTypes { // Rank- tensor of scalar type T. - typedef Eigen::TensorMap, - Eigen::Aligned> + typedef Eigen::TensorMap> Tensor; // Rank-1 tensor (vector) of scalar type T. - typedef Eigen::TensorMap, - Eigen::Aligned> - Vec; + typedef Eigen::TensorMap> Vec; typedef Eigen::TensorMap< Eigen::Tensor> ConstVec;