| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorContractionMapper.h | 75 typename nocontract_t, typename contract_t,
83 const contract_t& contract_strides,
84 const contract_t& k_strides) :
123 if (array_size<typename Tensor::Dimensions>::value > array_size<contract_t>::value) {
133 if(array_size<contract_t>::value > 0) {
134 for (int i = static_cast<int>(array_size<contract_t>::value) - 1; i > 0; i--) {
156 if (array_size<typename Tensor::Dimensions>::value > array_size<contract_t>::value) {
176 if (array_size<contract_t>::value> 0) {
177 for (int i = static_cast<int>(array_size<contract_t>::value) - 1; i > 0; i--) {
205 return ((side == Lhs) && inner_dim_contiguous && array_size<contract_t>::value > 0) ? m_contract_strides[0] : 1
[all...] |
| TensorContraction.h | 146 typedef array<Index, ContractDims> contract_t; typedef in struct:Eigen::TensorContractionEvaluatorBase
304 if (i+1 < static_cast<int>(internal::array_size<contract_t>::value)) {
397 contract_t, lhs_packet_size,
403 contract_t, rhs_packet_size,
453 contract_t, lhs_packet_size,
459 contract_t, rhs_packet_size,
546 contract_t m_k_strides;
547 contract_t m_left_contracting_strides;
548 contract_t m_right_contracting_strides;
603 typedef array<Index, ContractDims> contract_t; typedef in struct:Eigen::TensorEvaluator
[all...] |
| TensorContractionThreadPool.h | 96 typedef array<Index, ContractDims> contract_t; typedef in struct:Eigen::TensorEvaluator
130 contract_t, internal::packet_traits<LhsScalar>::size,
135 contract_t, internal::packet_traits<RhsScalar>::size,
762 contract_t, lhs_packet_size,
768 contract_t, rhs_packet_size,
[all...] |
| TensorContractionCuda.h | 1242 typedef array<Index, ContractDims> contract_t; typedef in struct:Eigen::TensorEvaluator
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| eigen_spatial_convolutions.h | 29 typename nocontract_t, typename contract_t, int Side, int packet_size,
37 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
47 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
56 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
69 const nocontract_t&, const nocontract_t&, const contract_t&,
70 const contract_t&)
238 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
458 typename nocontract_t, typename contract_t, int Side, int packet_size,
466 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
478 nocontract_t, contract_t, packet_size, inner_dim_contiguous
[all...] |
