Searched defs:TensorBlockMapper (Results 1 – 3 of 3) sorted by relevance
71 typedef internal::TensorBlockMapper<2, Layout> TensorBlockMapper; in test_block_mapper_sanity() typedef125 typedef internal::TensorBlockMapper<NumDims, Layout> TensorBlockMapper; in test_block_mapper_maps_every_element() typedef232 typedef internal::TensorBlockMapper<5, Layout> TensorBlockMapper; in test_uniform_block_shape() typedef359 typedef internal::TensorBlockMapper<5, Layout> TensorBlockMapper; in test_skewed_inner_dim_block_shape() typedef516 typedef internal::TensorBlockMapper<1, Layout> TensorBlockMapper; in test_empty_dims() typedef528 typedef internal::TensorBlockMapper<2, Layout> TensorBlockMapper; in test_empty_dims() typedef
76 using TensorBlockMapper = in test_block_io_copy_data_from_source_to_target() typedef147 using TensorBlockMapper = in test_block_io_copy_using_reordered_dimensions() typedef
354 TensorBlockMapper(const DSizes<IndexType, NumDims>& dimensions, in TensorBlockMapper() function