Searched refs:_GetTensorSpec (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
D | tf_trt_integration_test_base.py | 172 def _GetTensorSpec(self, shape, mask, dtype, name): member in TfTrtIntegrationTestBase 254 self._GetTensorSpec(shape, mask, dtype, "input_%d" % i) 258 self._GetTensorSpec(shape, mask, dtype, "output_%d" % i)
|