cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

No public description

Browse Source 
PiperOrigin-RevId: 561067189
This commit is contained in:
MediaPipe Team 2023-08-29 10:15:30 -07:00 committed by Copybara-Service
parent f56b8a13a3
commit 01fbbd9f67
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mediapipe/tasks/cc/vision/custom_ops/fused_batch_norm.cc
View File

@ -44,14 +44,11 @@ constexpr int kOutputSavedVarIndex = 4;
template <typename T, int NDIMS = 1, typename IndexType = Eigen::DenseIndex> template <typename T, int NDIMS = 1, typename IndexType = Eigen::DenseIndex>
struct TTypes { struct TTypes {
// Rank-<NDIMS> tensor of scalar type T. // Rank-<NDIMS> tensor of scalar type T.
typedef Eigen::TensorMap<Eigen::Tensor<T, NDIMS, Eigen::RowMajor, IndexType>, typedef Eigen::TensorMap<Eigen::Tensor<T, NDIMS, Eigen::RowMajor, IndexType>>
Eigen::Aligned>
Tensor; Tensor;
// Rank-1 tensor (vector) of scalar type T. // Rank-1 tensor (vector) of scalar type T.
typedef Eigen::TensorMap<Eigen::Tensor<T, 1, Eigen::RowMajor, IndexType>, typedef Eigen::TensorMap<Eigen::Tensor<T, 1, Eigen::RowMajor, IndexType>> Vec;
Eigen::Aligned>
Vec;
typedef Eigen::TensorMap< typedef Eigen::TensorMap<
Eigen::Tensor<const T, 1, Eigen::RowMajor, IndexType>> Eigen::Tensor<const T, 1, Eigen::RowMajor, IndexType>>
ConstVec; ConstVec;