Searched refs:TensorContractionInputMapper (Results 1 – 8 of 8) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorContraction.h | 395 typedef internal::TensorContractionInputMapper<LhsScalar, Index, internal::Lhs,
401 typedef internal::TensorContractionInputMapper<RhsScalar, Index, internal::Rhs,
451 typedef internal::TensorContractionInputMapper<LhsScalar, Index, internal::Lhs,
457 typedef internal::TensorContractionInputMapper<RhsScalar, Index, internal::Rhs,
|
| D | TensorContractionThreadPool.h | 128 typedef internal::TensorContractionInputMapper<
133 typedef internal::TensorContractionInputMapper<
760 typedef internal::TensorContractionInputMapper<LhsScalar, Index, internal::Lhs,
766 typedef internal::TensorContractionInputMapper<RhsScalar, Index, internal::Rhs,
|
| D | TensorContractionMapper.h | 436 class TensorContractionInputMapper
445 EIGEN_DEVICE_FUNC TensorContractionInputMapper(const Tensor& tensor,
|
| D | TensorContractionCuda.h | 1359 typedef internal::TensorContractionInputMapper<LhsScalar, Index, internal::Lhs,
1365 typedef internal::TensorContractionInputMapper<RhsScalar, Index, internal::Rhs,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_spatial_convolutions-inl.h | 61 class TensorContractionInputMapper<
72 typedef TensorContractionInputMapper<
99 TensorContractionInputMapper( in TensorContractionInputMapper() function
170 TensorContractionInputMapper(const TensorContractionInputMapper& base_mapper) in TensorContractionInputMapper() function
691 typedef TensorContractionInputMapper<
|
| D | eigen_cuboid_convolution.h | 64 class TensorContractionInputMapper<
74 typedef TensorContractionInputMapper<
97 TensorContractionInputMapper( in TensorContractionInputMapper() function
207 TensorContractionInputMapper(const TensorContractionInputMapper& base_mapper) in TensorContractionInputMapper() function
889 typedef TensorContractionInputMapper<
|
| D | eigen_contraction_kernel.h | 422 struct DirectColMajorAccess<TensorContractionInputMapper< \
428 using DataMapper = TensorContractionInputMapper< \
|
| D | eigen_spatial_convolutions_test.cc | 1422 using InputMapper = Eigen::internal::TensorContractionInputMapper< in PackRhsHelper()
1616 using InputMapper = Eigen::internal::TensorContractionInputMapper< in PackLhsHelper()
|