Searched refs:TensorContractionSubMapper (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_cuboid_convolution.h | 81 typedef TensorContractionSubMapper<
339 friend class TensorContractionSubMapper<
684 class TensorContractionSubMapper<
705 typedef TensorContractionSubMapper<
716 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorContractionSubMapper( in TensorContractionSubMapper() function
724 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorContractionSubMapper( in TensorContractionSubMapper() function
1007 TensorContractionSubMapper<
1016 typedef TensorContractionSubMapper<
1188 TensorContractionSubMapper<
1197 typedef TensorContractionSubMapper<
[all …]
|
| D | eigen_spatial_convolutions-inl.h | 80 typedef TensorContractionSubMapper<
271 friend class TensorContractionSubMapper<
514 class TensorContractionSubMapper<
536 typedef TensorContractionSubMapper<
548 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorContractionSubMapper( in TensorContractionSubMapper() function
556 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorContractionSubMapper( in TensorContractionSubMapper() function
809 TensorContractionSubMapper<
818 typedef TensorContractionSubMapper<
1016 TensorContractionSubMapper<
1025 typedef TensorContractionSubMapper<
[all …]
|
| D | eigen_spatial_convolutions.h | 37 TensorContractionSubMapper<
46 typedef TensorContractionSubMapper<
|
| D | eigen_spatial_convolutions_test.cc | 1429 using SubMapper = Eigen::internal::TensorContractionSubMapper< in PackRhsHelper()
1601 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…
|