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

Method get_shape_dict

demo/Diffusion/models.py:230–235  ·  view source on GitHub ↗
(self, batch_size, image_height, image_width)

Source from the content-addressed store, hash-verified

228 }
229
230 def get_shape_dict(self, batch_size, image_height, image_width):
231 self.check_dims(batch_size, image_height, image_width)
232 return {
233 'input_ids': (batch_size, self.text_maxlen),
234 'text_embeddings': (batch_size, self.text_maxlen, self.embedding_dim)
235 }
236
237 def get_sample_input(self, batch_size, image_height, image_width):
238 self.check_dims(batch_size, image_height, image_width)

Callers

nothing calls this directly

Calls 1

check_dimsMethod · 0.80

Tested by

no test coverage detected