| /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 | 62 typename nocontract_t, typename contract_t, int Side, int packet_size,
70 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
80 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
89 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
103 const nocontract_t&, const nocontract_t&, const contract_t&, in TensorContractionInputMapper() argument
104 const contract_t&) in TensorContractionInputMapper()
347 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
875 typename nocontract_t, typename contract_t, int Side, int packet_size,
883 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
895 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
[all …]
|
| D | eigen_spatial_convolutions-inl.h | 59 typename nocontract_t, typename contract_t, int Side, int packet_size,
67 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
78 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
88 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
104 const nocontract_t&, const nocontract_t&, const contract_t&, in TensorContractionInputMapper() argument
105 const contract_t&) in TensorContractionInputMapper()
283 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
677 typename nocontract_t, typename contract_t, int Side, int packet_size,
685 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
697 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
[all …]
|
| D | eigen_spatial_convolutions.h | 79 typename nocontract_t, typename contract_t, int packet_size,
89 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
98 nocontract_t, contract_t, packet_size, inner_dim_contiguous,
|
| D | eigen_spatial_convolutions_test.cc | 1409 using contract_t = Eigen::array<Eigen::Index, 1>; in PackRhsHelper() typedef
1424 nocontract_t, contract_t, // in PackRhsHelper()
1432 nocontract_t, contract_t, // in PackRhsHelper()
1457 contract_t contract_dim = {not_important}; in PackRhsHelper()
1604 using contract_t = Eigen::array<Eigen::Index, 1>; in PackLhsHelper() typedef
1618 nocontract_t, contract_t, // in PackLhsHelper()
1626 nocontract_t, contract_t, // in PackLhsHelper()
1655 contract_t contract_dim = {1}; in PackLhsHelper()
1660 contract_t contract_strides = {filter_count}; in PackLhsHelper()
1662 contract_t k_strides = {1}; in PackLhsHelper()
|
| D | eigen_contraction_kernel.h | 420 typename nocontract_t, typename contract_t, int packet_size, \
424 nocontract_t, contract_t, packet_size, /*inner_dim_contiguous=*/true, \
430 nocontract_t, contract_t, packet_size, /*inner_dim_contiguous=*/true, \
|