HomeSort by relevance Sort by last modified time
    Searched refs:TensorContractionBlocking (Results 1 - 3 of 3) sorted by null

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorContractionBlocking.h 25 class TensorContractionBlocking {
31 EIGEN_DEVICE_FUNC TensorContractionBlocking(Index k, Index m, Index n, Index num_threads = 1) :
TensorContractionThreadPool.h 184 internal::TensorContractionBlocking<LhsMapper, RhsMapper, Index,
191 internal::TensorContractionBlocking<LhsMapper, RhsMapper, Index,
228 internal::TensorContractionBlocking<LhsMapper, RhsMapper, Index,
235 internal::TensorContractionBlocking<LhsMapper, RhsMapper, Index,
    [all...]
TensorContraction.h 481 internal::TensorContractionBlocking<LhsMapper, RhsMapper, Index, internal::ShardByCol> blocking(k, m, n, 1);

Completed in 47 milliseconds