| /aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/ |
| H A D | _capture_strategies.py | 84 dynamic_shapes, argument
103 self, model, args, kwargs, dynamic_shapes argument
121 self, model, args, kwargs, dynamic_shapes argument
162 self, model, args, kwargs, dynamic_shapes argument
203 self, model, args, kwargs, dynamic_shapes argument
296 self, model, args, kwargs, dynamic_shapes argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_compile_op_common.cc | 99 OpInputList dynamic_shapes; in GetDynamicShapes() local
166 const std::vector<TensorShape>& dynamic_shapes, in CompileLocallyAndFillHostCache()
181 const std::vector<TensorShape>& dynamic_shapes, in CompileLocallyAndFillHostCacheInternal()
233 std::vector<TensorShape> dynamic_shapes; in ComputeInternal() local
|
| H A D | tpu_util.cc | 79 Status DynamicShapesToTensorShapes(const OpInputList& dynamic_shapes, in DynamicShapesToTensorShapes()
89 Status DynamicShapesToTensorShapes(const InputList& dynamic_shapes, in DynamicShapesToTensorShapes()
|
| H A D | tpu_op_util.cc | 30 const std::vector<tensorflow::TensorShape>& dynamic_shapes) { in CreateShapePrefix()
108 const std::vector<TensorShape>& dynamic_shapes, in CreateCompilationCacheKey()
|
| H A D | tpu_compile_op_common.h | 141 const std::vector<TensorShape>& dynamic_shapes, in LookupPersistentCompilationCacheAndFillCaches()
|
| H A D | tpu_compile_op_support.cc | 448 const std::vector<TensorShape>& dynamic_shapes, in ComputeArgumentShapes()
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_aot_inductor_utils.py | 31 dynamic_shapes=None, argument
120 dynamic_shapes=None, argument
139 dynamic_shapes=None, argument
|
| H A D | test_aot_inductor_package.py | 26 def compile(model, example_inputs, dynamic_shapes, options, device): argument
44 dynamic_shapes=None, argument
|
| H A D | indirect_assert_helper.py | 66 dynamic_shapes = dyn_shape == "True" variable
|
| /aosp_15_r20/external/executorch/backends/apple/mps/test/ |
| H A D | test_mps_utils.py | 193 dynamic_shapes=None, argument
298 dynamic_shapes=None, argument
320 dynamic_shapes=None, argument
|
| /aosp_15_r20/external/executorch/examples/models/llava/ |
| H A D | export_llava.py | 71 def export_text_model(llava, embeddings, dynamic_shapes): argument
123 def export_image_encoder(llava, resized, dynamic_shapes): argument
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | dynamic_shapes.py | 465 def f(path, t, *dynamic_shapes):
507 def _compare(tree, dynamic_shapes, path): argument
618 def dynamic_shapes(self, m, args, kwargs=None): member in ShapesCollection
|
| H A D | _trace.py | 1045 def _process_export_inputs(mod, args, kwargs, dynamic_shapes): argument
1105 export_artifact: ExportArtifact, combined_args: Dict[str, Any], dynamic_shapes argument
|
| /aosp_15_r20/external/pytorch/torch/_export/db/examples/ |
| H A D | dynamic_shape_round.py | 20 dynamic_shapes = {"x": {0: dim0_x}} variable
|
| H A D | scalar_output.py | 22 dynamic_shapes = {"x": {1: dim1_x}} variable
|
| H A D | cond_operands.py | 35 dynamic_shapes = {"x": {0: dim0_x}, "y": None} variable
|
| /aosp_15_r20/external/executorch/backends/xnnpack/test/models/ |
| H A D | mobilenet_v3.py | 19 dynamic_shapes = ( variable in TestMobileNetV3
|
| H A D | torchvision_vit.py | 18 dynamic_shapes = ( variable in TestViT
|
| H A D | resnet.py | 17 dynamic_shapes = ( variable in TestResNet18
|
| /aosp_15_r20/external/pytorch/torch/_export/db/ |
| H A D | case.py | 87 dynamic_shapes: Optional[Dict[str, Any]] = None variable in ExportCase
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | non_strict_utils.py | 118 dynamic_shapes, argument
|
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
| H A D | schemas.py | 814 dynamic_shapes: bool = False variable in AOTConfig
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | config.py | 67 dynamic_shapes = True variable
|
| /aosp_15_r20/external/executorch/backends/xnnpack/test/ops/ |
| H A D | linear.py | 553 dynamic_shapes, argument
|
| /aosp_15_r20/external/executorch/docs/source/tutorials_source/ |
| H A D | export-to-executorch-tutorial.py | 132 dynamic_shapes = {"x": {1: dim1_x}, "y": {1: dim1_x}} variable
|