Searched defs:dyn_shape (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/tests/ |
| H A D | raw_api_test.cc | 81 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in AcceptDynamicR1() local
91 xla::Shape dyn_shape; in AcceptDynamicR2() local
118 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in AcceptDynamicR1Tuple() local
1138 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in TEST() local
1194 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {2, 4}); in TEST() local
1247 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in TEST() local
1317 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in TEST() local
1374 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {4}); in TEST() local
1418 xla::Shape dyn_shape = xla::ShapeUtil::MakeShape(xla::F32, {2, 4}); in TEST() local
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | indirect_assert_helper.py | 54 _, fn_name, dims, dyn_shape, one_size = sys.argv variable
|
| H A D | test_torchinductor.py | 11175 def test(fn, ndims, dyn_shape, one_size=False): argument
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | fake_impls.py | 267 def dyn_shape(fake_mode, func, *args, **kwargs): function
|