Home | History | Annotate | Download | only in kernels

Lines Matching refs:expected

66   Tensor expected(allocator(), DT_FLOAT, {5});
67 test::FillValues<float>(&expected, {-2, 2, -2, 2, 2});
68 test::ExpectTensorEqual<float>(expected, *GetOutput(0));
85 Tensor expected(allocator(), DT_DOUBLE, {5});
86 test::FillValues<double>(&expected, {-2, 2, -2, 2, 2});
87 test::ExpectTensorEqual<double>(expected, *GetOutput(0));
104 Tensor expected(allocator(), DT_FLOAT, {5});
105 test::FillValues<float>(&expected, {-2, 3, -2, 4, 5});
106 test::ExpectTensorEqual<float>(expected, *GetOutput(0));
123 Tensor expected(allocator(), DT_FLOAT, {3, 4});
124 expected.flat<float>().setConstant(-2);
125 expected.tensor<float, 2>()(0, 1) = 2;
126 expected.tensor<float, 2>()(0, 2) = 2;
127 expected.tensor<float, 2>()(2, 3) = 2;
128 test::ExpectTensorEqual<float>(expected, *GetOutput(0));
145 Tensor expected(allocator(), DT_FLOAT, {3, 4});
146 expected.flat<float>().setConstant(-2);
147 expected.tensor<float, 2>()(0, 1) = 3;
148 expected.tensor<float, 2>()(0, 2) = 4;
149 expected.tensor<float, 2>()(2, 3) = 5;
150 test::ExpectTensorEqual<float>(expected, *GetOutput(0));
167 Tensor expected(allocator(), DT_FLOAT, {3, 4, 2});
168 expected.flat<float>().setConstant(-2);
169 expected.tensor<float, 3>()(0, 1, 1) = 2;
170 expected.tensor<float, 3>()(0, 2, 0) = 2;
171 expected.tensor<float, 3>()(2, 3, 1) = 2;
172 test::ExpectTensorEqual<float>(expected, *GetOutput(0));
189 Tensor expected(allocator(), DT_FLOAT, {3, 4, 2});
190 expected.flat<float>().setConstant(-2);
191 expected.tensor<float, 3>()(0, 1, 1) = 3;
192 expected.tensor<float, 3>()(0, 2, 0) = 4;
193 expected.tensor<float, 3>()(2, 3, 1) = 5;
194 test::ExpectTensorEqual<float>(expected, *GetOutput(0));