Home
last modified time | relevance | path

Searched refs:linearizedIndex (Results 1 – 3 of 3) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DFusionOnTensors.cpp619 Value linearizedIndex = fusedEntryBlock->addArgument(indexType); in fuseWithReshapeByExpansion() local
625 linearizedIndex = in fuseWithReshapeByExpansion()
626 linearizedIndex * std_constant_index(expandedDimExtent); in fuseWithReshapeByExpansion()
627 linearizedIndex = in fuseWithReshapeByExpansion()
628 linearizedIndex + fusedEntryBlock->addArgument(indexType); in fuseWithReshapeByExpansion()
630 argReplacements[i] = linearizedIndex; in fuseWithReshapeByExpansion()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorFixedSize.h82 return m_storage.data()[linearizedIndex(indices)]; in coeff()
114 return m_storage.data()[linearizedIndex(indices)]; in coeffRef()
376 EIGEN_STRONG_INLINE Index linearizedIndex(const array<Index, NumIndices>& indices) const in linearizedIndex() function
DTensor.h127 return m_storage.data()[linearizedIndex(indices)]; in coeff()
167 return m_storage.data()[linearizedIndex(indices)]; in coeffRef()
515 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index linearizedIndex(const array<Index, NumIndices>& indice… in linearizedIndex() function