Test an int32-typed input. On a GPU, int32 tensors will be placed in host memory.
(self)
| 203 | [[1, 2, 3], [4, 5, 6]], dtype=np.float32)]) |
| 204 | |
| 205 | def testInt32Input(self): |
| 206 | """Test an int32-typed input. |
| 207 | |
| 208 | On a GPU, int32 tensors will be placed in host memory. |
| 209 | """ |
| 210 | |
| 211 | def AddToSelf(x): |
| 212 | return math_ops.add(x, x) |
| 213 | |
| 214 | self._compare(AddToSelf, [np.array([7, 1, 3], dtype=np.int32)]) |
| 215 | |
| 216 | def testMandatoryConstantInput(self): |
| 217 | """Tests an operator that has a mandatory-constant shape input.""" |