| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | 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_s…
[all …]
|
| D | TensorContraction.h | 146 typedef array<Index, ContractDims> contract_t;
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;
|
| D | TensorContractionThreadPool.h | 96 typedef array<Index, ContractDims> contract_t;
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,
|
| D | TensorContractionCuda.h | 1242 typedef array<Index, ContractDims> contract_t;
1361 contract_t, 4,
1367 contract_t, 4,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_cuboid_convolution.h | 60 typename nocontract_t, typename contract_t, int Side, int packet_size,
68 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
78 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
87 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
101 const nocontract_t&, const nocontract_t&, const contract_t&, in TensorContractionInputMapper() argument
102 const contract_t&) in TensorContractionInputMapper()
345 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
682 typename nocontract_t, typename contract_t, int Side, int packet_size,
690 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
702 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
[all …]
|
| D | eigen_spatial_convolutions-inl.h | 57 typename nocontract_t, typename contract_t, int Side, int packet_size,
65 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
76 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
86 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
100 const nocontract_t&, const nocontract_t&, const contract_t&, in TensorContractionInputMapper() argument
101 const contract_t&) in TensorContractionInputMapper()
277 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
512 typename nocontract_t, typename contract_t, int Side, int packet_size,
520 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
532 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
[all …]
|
| D | eigen_spatial_convolutions.h | 33 typename nocontract_t, typename contract_t, int packet_size,
43 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
52 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
|
| D | eigen_spatial_convolutions_test.cc | 1408 using contract_t = Eigen::array<Eigen::Index, 1>; in PackRhsHelper() typedef
1423 nocontract_t, contract_t, // in PackRhsHelper()
1431 nocontract_t, contract_t, // in PackRhsHelper()
1455 contract_t contract_dim = {not_important}; in PackRhsHelper()
1581 using contract_t = Eigen::array<Eigen::Index, 1>; in PackLhsHelper() typedef
1595 nocontract_t, contract_t, // in PackLhsHelper()
1603 nocontract_t, contract_t, // in PackLhsHelper()
1632 contract_t contract_dim = {1}; in PackLhsHelper()
1637 contract_t contract_strides = {filter_count}; in PackLhsHelper()
1639 contract_t k_strides = {1}; in PackLhsHelper()
|