mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-06-04 18:54:00 +08:00
Fix cxx11_tensor_{block_access, reduction} tests
This commit is contained in:
parent
1dcf5a6ed8
commit
8a977c1f46
@ -820,8 +820,8 @@ static void test_block_cwise_binary_io_zero_strides() {
|
|||||||
left_strides[4] = 0;
|
left_strides[4] = 0;
|
||||||
|
|
||||||
DSizes<Index, 5> right_sizes = rnd_dims;
|
DSizes<Index, 5> right_sizes = rnd_dims;
|
||||||
right_sizes[1] = 0;
|
right_sizes[1] = 1;
|
||||||
right_sizes[3] = 0;
|
right_sizes[3] = 1;
|
||||||
|
|
||||||
DSizes<Index, 5> right_strides(ComputeStrides<Layout, 5>(right_sizes));
|
DSizes<Index, 5> right_strides(ComputeStrides<Layout, 5>(right_sizes));
|
||||||
right_strides[1] = 0;
|
right_strides[1] = 0;
|
||||||
|
@ -57,6 +57,8 @@ template <typename Scalar,int DataLayout>
|
|||||||
static void test_simple_reductions() {
|
static void test_simple_reductions() {
|
||||||
Tensor<Scalar, 4, DataLayout> tensor(2, 3, 5, 7);
|
Tensor<Scalar, 4, DataLayout> tensor(2, 3, 5, 7);
|
||||||
tensor.setRandom();
|
tensor.setRandom();
|
||||||
|
// Add a little offset so that the product reductions won't be close to zero.
|
||||||
|
tensor += tensor.constant(Scalar(0.5f));
|
||||||
array<ptrdiff_t, 2> reduction_axis2;
|
array<ptrdiff_t, 2> reduction_axis2;
|
||||||
reduction_axis2[0] = 1;
|
reduction_axis2[0] = 1;
|
||||||
reduction_axis2[1] = 3;
|
reduction_axis2[1] = 3;
|
||||||
|
Loading…
x
Reference in New Issue
Block a user