Searched refs:TensorContractionInputMapper (Results 1 – 10 of 10) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorContractionMapper.h | 524 class TensorContractionInputMapper
533 EIGEN_DEVICE_FUNC TensorContractionInputMapper(const Tensor& tensor,
562 struct TensorContractionInputMapperTrait<TensorContractionInputMapper<Scalar_, Index_, side_, Tenso…
|
| D | TensorContraction.h | 744 typedef internal::TensorContractionInputMapper<LhsScalar, Index, internal::Lhs,
750 typedef internal::TensorContractionInputMapper<RhsScalar, Index, internal::Rhs,
822 typedef internal::TensorContractionInputMapper<LhsScalar, Index, internal::Lhs,
828 typedef internal::TensorContractionInputMapper<RhsScalar, Index, internal::Rhs,
|
| D | TensorContractionThreadPool.h | 354 typedef internal::TensorContractionInputMapper<
359 typedef internal::TensorContractionInputMapper<
|
| D | TensorContractionSycl.h | 1393 typedef internal::TensorContractionInputMapper<
1398 …typedef internal::TensorContractionInputMapper<RhsScalar, StorageIndex, internal::Rhs, RightEvalua…
|
| D | TensorContractionGpu.h | 1376 typedef internal::TensorContractionInputMapper<LhsScalar, Index, internal::Lhs,
1382 typedef internal::TensorContractionInputMapper<RhsScalar, Index, internal::Rhs,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_spatial_convolutions-inl.h | 62 class TensorContractionInputMapper<
73 typedef TensorContractionInputMapper<
100 TensorContractionInputMapper( in TensorContractionInputMapper() function
171 TensorContractionInputMapper(const TensorContractionInputMapper& base_mapper) in TensorContractionInputMapper() function
692 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()
|
| /external/tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/ |
| D | MatMatProductAVX2.h | 542 TensorContractionInputMapper<
545 TensorContractionInputMapper<QUInt8, Index, Rhs, RightTensor,
|