Home
last modified time | relevance | path

Searched refs:_GetTensorSpec (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py172 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)