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

Method get_input_profile

demo/Diffusion/models.py:223–228  ·  view source on GitHub ↗
(self, batch_size, image_height, image_width, static_batch, static_shape)

Source from the content-addressed store, hash-verified

221 }
222
223 def get_input_profile(self, batch_size, image_height, image_width, static_batch, static_shape):
224 self.check_dims(batch_size, image_height, image_width)
225 min_batch, max_batch, _, _, _, _, _, _, _, _ = self.get_minmax_dims(batch_size, image_height, image_width, static_batch, static_shape)
226 return {
227 'input_ids': [(min_batch, self.text_maxlen), (batch_size, self.text_maxlen), (max_batch, self.text_maxlen)]
228 }
229
230 def get_shape_dict(self, batch_size, image_height, image_width):
231 self.check_dims(batch_size, image_height, image_width)

Callers

nothing calls this directly

Calls 2

check_dimsMethod · 0.80
get_minmax_dimsMethod · 0.80

Tested by

no test coverage detected