MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / testNumbersCPU

Method testNumbersCPU

tensorflow/python/kernel_tests/relu_op_test.py:261–266  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

259 self.assertShapeEqual(np_relu6, tf_relu6)
260
261 def testNumbersCPU(self):
262 for t in [np.int32, np.int64, np.float16, np.float32, np.float64]:
263 # Force execution on CPU even if a GPU kernel is available for the type.
264 with ops.device("/device:CPU:0"):
265 self._testRelu6(
266 np.array([[-9, 7, -5, 3, -1], [1, -3, 5, -7, 9]]).astype(t))
267
268 def testNumbersGPU(self):
269 if not test.is_gpu_available():

Callers

nothing calls this directly

Calls 2

_testRelu6Method · 0.95
deviceMethod · 0.45

Tested by

no test coverage detected