Home | History | Annotate | Download | only in test

Lines Matching refs:broadcast

28   no_broadcast = tensor.broadcast(broadcasts);
49 Tensor<float, 4, DataLayout> broadcast;
50 broadcast = tensor.broadcast(broadcasts);
52 VERIFY_IS_EQUAL(broadcast.dimension(0), 4);
53 VERIFY_IS_EQUAL(broadcast.dimension(1), 9);
54 VERIFY_IS_EQUAL(broadcast.dimension(2), 5);
55 VERIFY_IS_EQUAL(broadcast.dimension(3), 28);
61 VERIFY_IS_EQUAL(tensor(i%2,j%3,k%5,l%7), broadcast(i,j,k,l));
79 Tensor<float, 3, DataLayout> broadcast;
80 broadcast = tensor.broadcast(broadcasts);
82 VERIFY_IS_EQUAL(broadcast.dimension(0), 16);
83 VERIFY_IS_EQUAL(broadcast.dimension(1), 9);
84 VERIFY_IS_EQUAL(broadcast.dimension(2), 20);
89 VERIFY_IS_EQUAL(tensor(i%8,j%3,k%5), broadcast(i,j,k));
96 broadcast = tensor.broadcast(broadcasts);
98 VERIFY_IS_EQUAL(broadcast.dimension(0), 22);
99 VERIFY_IS_EQUAL(broadcast.dimension(1), 9);
100 VERIFY_IS_EQUAL(broadcast.dimension(2), 20);
105 VERIFY_IS_EQUAL(tensor(i%11,j%3,k%5), broadcast(i,j,k));
127 Tensor<float, 3, DataLayout> broadcast;
128 broadcast = tensor.broadcast(broadcasts);
130 VERIFY_IS_EQUAL(broadcast.dimension(0), 16);
131 VERIFY_IS_EQUAL(broadcast.dimension(1), 9);
132 VERIFY_IS_EQUAL(broadcast.dimension(2), 20);
137 VERIFY_IS_EQUAL(tensor(i%8,j%3,k%5), broadcast(i,j,k));
144 broadcast = tensor.broadcast(broadcasts);
146 VERIFY_IS_EQUAL(broadcast.dimension(0), 22);
147 VERIFY_IS_EQUAL(broadcast.dimension(1), 9);
148 VERIFY_IS_EQUAL(broadcast.dimension(2), 20);
153 VERIFY_IS_EQUAL(tensor(i%11,j%3,k%5), broadcast(i,j,k));
170 Tensor<float, 1, DataLayout> t3 = t1 + t2.broadcast(Eigen::array<int, 1>{{10}});
176 Tensor<float, 1, DataLayout> t5 = t1 + t4.broadcast(Eigen::array<int, 1>{{10}});