Searched refs:TensorContractionSubMapper (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A 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<
1392 TensorContractionSubMapper<
1401 typedef TensorContractionSubMapper<
[all …]
|
| H A D | eigen_spatial_convolutions-inl.h | 83 typedef TensorContractionSubMapper<
278 friend class TensorContractionSubMapper<
680 class TensorContractionSubMapper<
702 typedef TensorContractionSubMapper<
716 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorContractionSubMapper( in TensorContractionSubMapper() function
724 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorContractionSubMapper( in TensorContractionSubMapper() function
1047 TensorContractionSubMapper<
1056 typedef TensorContractionSubMapper<
1264 TensorContractionSubMapper<
1273 typedef TensorContractionSubMapper<
[all …]
|
| H A D | eigen_spatial_convolutions.h | 83 TensorContractionSubMapper<
92 typedef TensorContractionSubMapper<
|
| H A D | eigen_spatial_convolutions_test.cc | 1430 using SubMapper = Eigen::internal::TensorContractionSubMapper< in PackRhsHelper()
1624 using SubMapper = Eigen::internal::TensorContractionSubMapper< in PackLhsHelper()
|
| /aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| H A D | TensorContractionMapper.h | 411 class TensorContractionSubMapper {
415 …typedef TensorContractionSubMapper<Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_s…
424 …EIGEN_DEVICE_FUNC TensorContractionSubMapper(const ParentMapper& base_mapper, Index vert_offset, I…
530 …typedef TensorContractionSubMapper<Scalar, Index, side, Tensor, nocontract_t, contract_t, packet_s…
|