Searched defs:return_shapes (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/backends/_nnapi/ |
| H A D | prepare.py | 101 return_shapes=None, argument
157 model, inputs, serializer=None, return_shapes=None, use_int16_for_qint16=False argument
|
| H A D | serializer.py | 704 def serialize_model(self, model, inputs, return_shapes=None): argument
2214 module, inputs, *, config=None, return_shapes=None, use_int16_for_qint16=False argument
|