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

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorContraction.h 143 static const int ContractDims = internal::array_size<Indices>::value;
144 static const int NumDims = LDims + RDims - 2 * ContractDims;
146 typedef array<Index, ContractDims> contract_t;
147 typedef array<Index, LDims - ContractDims> left_nocontract_t;
148 typedef array<Index, RDims - ContractDims> right_nocontract_t;
167 array<IndexPair<Index>, ContractDims> eval_op_indices;
177 for (int i = 0; i < ContractDims; i++) {
191 for (int i = 0; i < ContractDims; i++) {
192 eval_op_indices[i].first = LDims - 1 - op.indices()[ContractDims - 1 - i].second;
193 eval_op_indices[i].second = RDims - 1 - op.indices()[ContractDims - 1 - i].first
    [all...]
TensorContractionThreadPool.h 91 static const int ContractDims = internal::array_size<Indices>::value;
96 typedef array<Index, ContractDims> contract_t;
97 typedef array<Index, LDims - ContractDims> left_nocontract_t;
98 typedef array<Index, RDims - ContractDims> right_nocontract_t;
100 static const int NumDims = LDims + RDims - 2 * ContractDims;
    [all...]
TensorContractionCuda.h     [all...]

Completed in 326 milliseconds