Searched refs:block_sizes (Results 1 – 4 of 4) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_block_io.cpp | 371 DSizes<Index, 5> block_sizes(1, 2, 1, 2, 1); in test_block_io_squeeze_ones() local
372 auto strides = internal::strides<Layout>(block_sizes); in test_block_io_squeeze_ones()
375 Tensor<float, 5> input(block_sizes); in test_block_io_squeeze_ones()
378 Tensor<float, 5> output(block_sizes); in test_block_io_squeeze_ones()
380 IODst dst(block_sizes, strides, output.data(), 0); in test_block_io_squeeze_ones()
384 for (Index i = 0; i < block_sizes.TotalSize(); ++i) { in test_block_io_squeeze_ones()
391 DSizes<Index, 5> block_sizes(1, 1, 1, 1, 1); in test_block_io_squeeze_ones() local
392 auto strides = internal::strides<Layout>(block_sizes); in test_block_io_squeeze_ones()
395 Tensor<float, 5> input(block_sizes); in test_block_io_squeeze_ones()
398 Tensor<float, 5> output(block_sizes); in test_block_io_squeeze_ones()
[all …]
|
| D | cxx11_tensor_block_access.cpp | 107 const DSizes<Index, NumDims>& block_sizes = block.dimensions(); in UpdateCoeffSet() local
109 for (int i = 0; i < block_sizes[dim_index]; ++i) { in UpdateCoeffSet()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_util.py | 581 block_sizes = [dim.value for dim in block_dims]
582 if any(d is None for d in block_sizes):
583 block_sizes = block_dims_fn()
584 return array_ops.split(arg, block_sizes, axis=axis)
|
| /external/python/cpython3/Lib/test/ |
| D | test_urllibnet.py | 222 block_sizes = {block_size for _, block_size, _ in records}
223 self.assertEqual({records[0][1]}, block_sizes,
|