Searched refs:TensorContractionInputMapper (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| H A D | TensorContractionMapper.h | 524 class TensorContractionInputMapper
533 EIGEN_DEVICE_FUNC TensorContractionInputMapper(const Tensor& tensor,
562 struct TensorContractionInputMapperTrait<TensorContractionInputMapper<Scalar_, Index_, side_, Tenso…
|
| H A 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,
|
| H A D | TensorContractionThreadPool.h | 354 typedef internal::TensorContractionInputMapper<
359 typedef internal::TensorContractionInputMapper<
|
| H A D | TensorContractionSycl.h | 1393 typedef internal::TensorContractionInputMapper<
1398 …typedef internal::TensorContractionInputMapper<RhsScalar, StorageIndex, internal::Rhs, RightEvalua…
|
| H A D | TensorContractionGpu.h | 1376 typedef internal::TensorContractionInputMapper<LhsScalar, Index, internal::Lhs,
1382 typedef internal::TensorContractionInputMapper<RhsScalar, Index, internal::Rhs,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A 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<
|
| H A 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<
|
| H A D | eigen_contraction_kernel.h | 422 struct DirectColMajorAccess<TensorContractionInputMapper< \
428 using DataMapper = TensorContractionInputMapper< \
|
| H A D | eigen_spatial_convolutions_test.cc | 1422 using InputMapper = Eigen::internal::TensorContractionInputMapper< in PackRhsHelper()
1616 using InputMapper = Eigen::internal::TensorContractionInputMapper< in PackLhsHelper()
|
| /aosp_15_r20/external/tensorflow/third_party/eigen3/unsupported/Eigen/CXX11/src/FixedPoint/ |
| H A D | MatMatProductAVX2.h | 542 TensorContractionInputMapper<
545 TensorContractionInputMapper<QUInt8, Index, Rhs, RightTensor,
|
| /aosp_15_r20/out/soong/.intermediates/packages/modules/NeuralNetworks/common/random/philox_random/android_arm64_armv8-2a_cortex-a55_static_apex30_sabi/obj/packages/modules/NeuralNetworks/common/random/ |
| D | simple_philox.sdump | 98349 …"name" : "Eigen::internal::TensorContractionInputMapperTrait<TensorContractionInputMapper<type-par…
98355 …"name" : "Eigen::internal::TensorContractionInputMapperTrait<TensorContractionInputMapper<type-par…
|