| 122 | } |
| 123 | |
| 124 | void TestBasic() { |
| 125 | Tensor input_tensor(DT_QUINT8, {1, 4, 4, 32}); |
| 126 | auto input = input_tensor.flat<quint8>(); |
| 127 | // Random input |
| 128 | input = input.random(Eigen::internal::UniformRandomGenerator<quint8>()); |
| 129 | |
| 130 | Expect(input_tensor, 0.0f, 1.0f, false, 0.0f, 0.0f); |
| 131 | } |
| 132 | |
| 133 | void TestZeroInput() { |
| 134 | Tensor input_tensor(DT_QUINT8, {1, 4, 4, 32}); |