Home
last modified time | relevance | path

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

/external/eigen/unsupported/test/
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 …]
Dcxx11_tensor_block_io.cpp25 static internal::TensorBlockShapeType RandomBlockShape() { in RandomBlockShape()
27 ? internal::TensorBlockShapeType::kUniformAllDims in RandomBlockShape()
28 : internal::TensorBlockShapeType::kSkewedInnerDims; in RandomBlockShape()
Dcxx11_tensor_block_eval.cpp66 {internal::TensorBlockShapeType::kSkewedInnerDims, in SkewedInnerBlock()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
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 …]