Home
last modified time | relevance | path

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

/external/eigen/unsupported/test/
Dcxx11_tensor_block_io.cpp63 using TensorBlockIO = internal::TensorBlockIO<T, Index, NumDims, Layout>; in test_block_io_copy_data_from_source_to_target() typedef
64 using IODst = typename TensorBlockIO::Dst; in test_block_io_copy_data_from_source_to_target()
65 using IOSrc = typename TensorBlockIO::Src; in test_block_io_copy_data_from_source_to_target()
103 TensorBlockIO::Copy(dst, src); in test_block_io_copy_data_from_source_to_target()
111 TensorBlockIO::Copy(dst, src); in test_block_io_copy_data_from_source_to_target()
174 using TensorBlockIO = internal::TensorBlockIO<T, Index, NumDims, Layout>; in test_block_io_copy_using_reordered_dimensions() typedef
175 using IODst = typename TensorBlockIO::Dst; in test_block_io_copy_using_reordered_dimensions()
176 using IOSrc = typename TensorBlockIO::Src; in test_block_io_copy_using_reordered_dimensions()
190 TensorBlockIO::Copy(dst, src, /*dst_to_src_dim_map=*/dim_map); in test_block_io_copy_using_reordered_dimensions()
208 TensorBlockIO::Copy(dst, src, /*dst_to_src_dim_map=*/dim_map); in test_block_io_copy_using_reordered_dimensions()
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorShuffling.h260 typedef internal::TensorBlockIO<ScalarNoConst, Index, NumDims, Layout>
261 TensorBlockIO;
262 typedef typename TensorBlockIO::Dst TensorBlockIODst;
263 typedef typename TensorBlockIO::Src TensorBlockIOSrc;
269 typename TensorBlockIO::Dimensions input_strides(m_unshuffledInputStrides);
275 typename TensorBlockIO::DimensionsMap dst_to_src_dim_map(m_shuffle);
276 TensorBlockIO::Copy(dst, src, dst_to_src_dim_map);
415 typedef internal::TensorBlockIO<ScalarNoConst, Index, NumDims, Layout>
416 TensorBlockIO;
417 typedef typename TensorBlockIO::Dst TensorBlockIODst;
[all …]
DTensorBlock.h17 class TensorBlockIO; variable
807 typedef internal::TensorBlockIO<Scalar, IndexType, NumDims, Layout>
808 TensorBlockIO;
809 typedef typename TensorBlockIO::Dst TensorBlockIODst;
810 typedef typename TensorBlockIO::Src TensorBlockIOSrc;
817 TensorBlockIO::Copy(dst, src);
1143 class TensorBlockIO {
DTensorBroadcasting.h1071 typedef internal::TensorBlockIO<ScalarNoConst, Index, 2 * NumDims, Layout>
1072 TensorBlockIO;
1074 typename TensorBlockIO::Src src(bcast_input_strides, input_buffer);
1075 typename TensorBlockIO::Dst dst(bcast_block_sizes, bcast_block_strides,
1078 return TensorBlockIO::Copy(dst, src);