Searched defs:get_dynamic_shapes (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/executorch/examples/models/llama3_2_vision/text_decoder/ | ||
| H A D | model.py | 185 def get_dynamic_shapes(self): member in Llama3_2Decoder |
| /aosp_15_r20/external/executorch/examples/models/llama3_2_vision/preprocess/ | ||
| H A D | model.py | 63 def get_dynamic_shapes(self) -> Dict[str, Dict[int, Dim]]: member in CLIPImageTransformModel |
| /aosp_15_r20/external/executorch/examples/models/llama3_2_vision/vision_encoder/ | ||
| H A D | model.py | 84 def get_dynamic_shapes(self): member in FlamingoVisionEncoderModel |
| /aosp_15_r20/external/executorch/examples/models/llava/ | ||
| H A D | model.py | 322 def get_dynamic_shapes(self): member in LlavaModel |
| /aosp_15_r20/external/executorch/test/models/ | ||
| H A D | export_program.py | 119 def get_dynamic_shapes(self): member in ModuleDynamicCatUnallocatedIO |
| /aosp_15_r20/external/executorch/exir/program/test/ | ||
| H A D | test_program.py | 367 def get_dynamic_shapes(self): member in TestProgramManagers.test_issue_3659.Mul |