Searched refs:TensorContractionSubMapper (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_cuboid_convolution.h | 83 typedef TensorContractionSubMapper<
341 friend class TensorContractionSubMapper<
877 class TensorContractionSubMapper<
898 typedef TensorContractionSubMapper<
909 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorContractionSubMapper( in TensorContractionSubMapper() function
917 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorContractionSubMapper( in TensorContractionSubMapper() function
1201 TensorContractionSubMapper<
1210 typedef TensorContractionSubMapper<
1382 TensorContractionSubMapper<
1391 typedef TensorContractionSubMapper<
[all …]
|
| D | eigen_spatial_convolutions-inl.h | 82 typedef TensorContractionSubMapper<
277 friend class TensorContractionSubMapper<
679 class TensorContractionSubMapper<
701 typedef TensorContractionSubMapper<
715 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorContractionSubMapper( in TensorContractionSubMapper() function
723 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorContractionSubMapper( in TensorContractionSubMapper() function
1046 TensorContractionSubMapper<
1055 typedef TensorContractionSubMapper<
1253 TensorContractionSubMapper<
1262 typedef TensorContractionSubMapper<
[all …]
|
| D | eigen_spatial_convolutions.h | 83 TensorContractionSubMapper<
92 typedef TensorContractionSubMapper<
|
| D | eigen_spatial_convolutions_test.cc | 1430 using SubMapper = Eigen::internal::TensorContractionSubMapper< in PackRhsHelper()
1624 using SubMapper = Eigen::internal::TensorContractionSubMapper< in PackLhsHelper()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorContractionMapper.h | 337 class TensorContractionSubMapper {
343 …typedef TensorContractionSubMapper<Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_s…
352 …EIGEN_DEVICE_FUNC TensorContractionSubMapper(const ParentMapper& base_mapper, Index vert_offset, I…
442 …typedef TensorContractionSubMapper<Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_s…
|