Searched refs:assert_jit_shape_analysis (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_ops_jit.py | 170 if op.assert_jit_shape_analysis:
218 sizes, traced_fn.graph, op.assert_jit_shape_analysis
221 if op.assert_jit_shape_analysis:
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 12211 assert_jit_shape_analysis=True,
12647 assert_jit_shape_analysis=True,
13069 assert_jit_shape_analysis=True,
13089 assert_jit_shape_analysis=True,
13956 assert_jit_shape_analysis=True,
14365 assert_jit_shape_analysis=True,
14402 assert_jit_shape_analysis=False,
14571 assert_jit_shape_analysis=True,
14592 assert_jit_shape_analysis=True,
14621 assert_jit_shape_analysis=True,
[all …]
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | functorch_additional_op_db.py | 386 assert_jit_shape_analysis=False, # TODO: support index.Tensor()
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/ |
| H A D | core.py | 891 assert_jit_shape_analysis: bool = False variable in OpInfo
|