Searched refs:TensorBlockMapper (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_block_access.cpp | 71 typedef internal::TensorBlockMapper<2, Layout> TensorBlockMapper; in test_block_mapper_sanity() typedef
76 TensorBlockMapper uniform_block_mapper( in test_block_mapper_sanity()
88 TensorBlockMapper skewed_block_mapper( in test_block_mapper_sanity()
125 typedef internal::TensorBlockMapper<NumDims, Layout> TensorBlockMapper; in test_block_mapper_maps_every_element() typedef
134 TensorBlockMapper block_mapper( in test_block_mapper_maps_every_element()
232 typedef internal::TensorBlockMapper<5, Layout> TensorBlockMapper; in test_uniform_block_shape() typedef
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 D | cxx11_tensor_block_io.cpp | 76 using TensorBlockMapper = in test_block_io_copy_data_from_source_to_target() typedef
77 internal::TensorBlockMapper<NumDims, Layout, Index>; in test_block_io_copy_data_from_source_to_target()
78 TensorBlockMapper block_mapper( in test_block_io_copy_data_from_source_to_target()
147 using TensorBlockMapper = in test_block_io_copy_using_reordered_dimensions() typedef
148 internal::TensorBlockMapper<NumDims, Layout, Index>; in test_block_io_copy_using_reordered_dimensions()
149 TensorBlockMapper block_mapper(output_tensor_dims, in test_block_io_copy_using_reordered_dimensions()
|
| H A D | cxx11_tensor_block_eval.cpp | 64 using BlockMapper = internal::TensorBlockMapper<NumDims, Layout, Index>; in SkewedInnerBlock()
|
| /aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| H A D | TensorExecutor.h | 178 typedef TensorBlockMapper<NumDims, Evaluator::Layout, StorageIndex>
179 TensorBlockMapper;
196 const TensorBlockMapper block_mapper(
227 template <typename TensorBlockMapper>
230 TensorExecutorTilingContext(const TensorBlockMapper& b_mapper,
236 TensorBlockMapper block_mapper; // navigate through blocks
243 template <typename Evaluator, typename TensorBlockMapper, bool Vectorizable>
244 TensorExecutorTilingContext<TensorBlockMapper> GetTensorExecutorTilingContext(
255 TensorBlockMapper block_mapper(
256 typename TensorBlockMapper::Dimensions(evaluator.dimensions()),
[all …]
|
| H A D | TensorBlock.h | 347 class TensorBlockMapper {
353 TensorBlockMapper() = default;
354 TensorBlockMapper(const DSizes<IndexType, NumDims>& dimensions, in TensorBlockMapper() function
|