Home
last modified time | relevance | path

Searched refs:TensorBlockShapeType (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_block_access.cpp22 using Eigen::internal::TensorBlockShapeType;
31 static TensorBlockShapeType RandomShape() { in RandomShape()
33 ? TensorBlockShapeType::kUniformAllDims in RandomShape()
34 : TensorBlockShapeType::kSkewedInnerDims; in RandomShape()
77 tensor_dims, {TensorBlockShapeType::kUniformAllDims, 100, zeroCost()}); in test_block_mapper_sanity()
89 tensor_dims, {TensorBlockShapeType::kSkewedInnerDims, 100, zeroCost()}); in test_block_mapper_sanity()
238 TensorBlockMapper block_mapper(dims, {TensorBlockShapeType::kUniformAllDims, in test_uniform_block_shape()
252 TensorBlockMapper block_mapper(dims, {TensorBlockShapeType::kUniformAllDims, in test_uniform_block_shape()
263 TensorBlockMapper block_mapper(dims, {TensorBlockShapeType::kUniformAllDims, in test_uniform_block_shape()
278 TensorBlockMapper block_mapper(dims, {TensorBlockShapeType::kUniformAllDims, in test_uniform_block_shape()
[all …]
H A Dcxx11_tensor_block_io.cpp25 static internal::TensorBlockShapeType RandomBlockShape() { in RandomBlockShape()
27 ? internal::TensorBlockShapeType::kUniformAllDims in RandomBlockShape()
28 : internal::TensorBlockShapeType::kSkewedInnerDims; in RandomBlockShape()
H A Dcxx11_tensor_block_eval.cpp66 {internal::TensorBlockShapeType::kSkewedInnerDims, in SkewedInnerBlock()
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorBlock.h73 enum class TensorBlockShapeType { kUniformAllDims, kSkewedInnerDims }; enum
76 TensorBlockShapeType shape_type; // target block shape
85 TensorBlockResourceRequirements(TensorBlockShapeType shape_type_, size_t size_, in TensorBlockResourceRequirements()
93 TensorBlockShapeType shape_type, size_t size_in_bytes, in withShapeAndSize()
101 TensorBlockShapeType shape_type, size_t size_in_bytes) { in withShapeAndSize()
125 return withShapeAndSize<Scalar>(TensorBlockShapeType::kSkewedInnerDims, in skewed()
132 return withShapeAndSize<Scalar>(TensorBlockShapeType::kUniformAllDims, in uniform()
156 return {TensorBlockShapeType::kUniformAllDims, 1, {0, 0, 0}}; in any()
168 static EIGEN_STRONG_INLINE TensorBlockShapeType
169 merge(TensorBlockShapeType lhs, TensorBlockShapeType rhs) { in merge()
[all …]
/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/
Dsimple_philox.sdump8228 "name" : "Eigen::internal::TensorBlockShapeType::kUniformAllDims"
8232 "name" : "Eigen::internal::TensorBlockShapeType::kSkewedInnerDims"
8236 "name" : "Eigen::internal::TensorBlockShapeType",