MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / input_spec

Method input_spec

samples/python/efficientdet/infer_tf.py:65–66  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

63 self.batch_size = shape[0]
64
65 def input_spec(self):
66 return self.inputs[0]['shape'], self.inputs[0]['dtype']
67
68 def output_spec(self):
69 return self.outputs[0]['shape'], self.outputs[0]['dtype']

Callers 2

mainFunction · 0.95
mainFunction · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected