Searched refs:block_to_tensor_dim (Results 1 – 1 of 1) sorted by relevance
225 DSizes<int, 3> block_to_tensor_dim; in test_block_io_copy_using_reordered_dimensions_do_not_squeeze() local226 block_to_tensor_dim[0] = 2; in test_block_io_copy_using_reordered_dimensions_do_not_squeeze()227 block_to_tensor_dim[1] = 1; in test_block_io_copy_using_reordered_dimensions_do_not_squeeze()228 block_to_tensor_dim[2] = 0; in test_block_io_copy_using_reordered_dimensions_do_not_squeeze()248 TensorBlockIO::Copy(dst, src, /*dst_to_src_dim_map=*/block_to_tensor_dim); in test_block_io_copy_using_reordered_dimensions_do_not_squeeze()272 DSizes<int, 4> block_to_tensor_dim; in test_block_io_copy_using_reordered_dimensions_squeeze() local273 block_to_tensor_dim[0] = 0; in test_block_io_copy_using_reordered_dimensions_squeeze()274 block_to_tensor_dim[1] = 1; in test_block_io_copy_using_reordered_dimensions_squeeze()275 block_to_tensor_dim[2] = 3; in test_block_io_copy_using_reordered_dimensions_squeeze()276 block_to_tensor_dim[3] = 2; in test_block_io_copy_using_reordered_dimensions_squeeze()[all …]