| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorContractionMapper.h | 75 typename nocontract_t, typename contract_t,
81 const nocontract_t& nocontract_strides,
82 const nocontract_t& ij_strides,
118 for (int i = static_cast<int>(array_size<nocontract_t>::value) - 1; i > 0; i--) {
157 for (int i = static_cast<int>(array_size<nocontract_t>::value) - 1; i > 0; i--) {
210 const nocontract_t m_nocontract_strides;
211 const nocontract_t m_ij_strides;
219 typename nocontract_t, typename contract_t,
222 …r : public SimpleTensorContractionMapper<Scalar, Index, side, Tensor, nocontract_t, contract_t, pa…
225 …typedef SimpleTensorContractionMapper<Scalar, Index, side, Tensor, nocontract_t, contract_t, packe…
[all …]
|
| /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
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,
904 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
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,
707 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 | 1408 using nocontract_t = Eigen::array<Eigen::Index, 1>; in PackRhsHelper() typedef
1424 nocontract_t, contract_t, // in PackRhsHelper()
1432 nocontract_t, contract_t, // in PackRhsHelper()
1456 nocontract_t nocontract_dim = {not_important}; in PackRhsHelper()
1603 using nocontract_t = Eigen::array<Eigen::Index, 1>; in PackLhsHelper() typedef
1618 nocontract_t, contract_t, // in PackLhsHelper()
1626 nocontract_t, contract_t, // in PackLhsHelper()
1654 nocontract_t nocontract_dim = {0}; in PackLhsHelper()
1659 nocontract_t nocontract_strides = {1}; in PackLhsHelper()
1661 nocontract_t i_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, \
|