Home
last modified time | relevance | path

Searched refs:allow_dynamic_output_shape_ops (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfake_impls.py276 or not fake_mode.shape_env.allow_dynamic_output_shape_ops
362 or not fake_mode.shape_env.allow_dynamic_output_shape_ops
403 or not fake_mode.shape_env.allow_dynamic_output_shape_ops
444 or not fake_mode.shape_env.allow_dynamic_output_shape_ops
725 or not fake_mode.shape_env.allow_dynamic_output_shape_ops
/aosp_15_r20/external/pytorch/torch/_library/
H A Dfake_impl.py185 or not self._shape_env.allow_dynamic_output_shape_ops
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dsymbolic_shapes.py2302 allow_dynamic_output_shape_ops: bool
2381 allow_dynamic_output_shape_ops=True, argument
2424 allow_dynamic_output_shape_ops=allow_dynamic_output_shape_ops,
2597 def allow_dynamic_output_shape_ops(self): member in ShapeEnv
2598 return self.settings.allow_dynamic_output_shape_ops
/aosp_15_r20/external/pytorch/torch/onnx/_internal/
H A D_exporter_legacy.py442 allow_scalar_outputs=False, allow_dynamic_output_shape_ops=False
/aosp_15_r20/external/pytorch/test/
H A Dtest_ops.py2435 allow_dynamic_output_shape_ops=True
2515 or not fake_mode.shape_env.allow_dynamic_output_shape_ops
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Doutput_graph.py299 allow_dynamic_output_shape_ops=config.capture_dynamic_output_shape_ops,
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_export.py3487 allow_dynamic_output_shape_ops=config.capture_dynamic_output_shape_ops,
3731 allow_dynamic_output_shape_ops=config.capture_dynamic_output_shape_ops,