Searched refs:testcase_shapes (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/dlpack/ |
| H A D | dlpack_test.py | 37 testcase_shapes = [(), (1,), (2, 3), (2, 0), (0, 7), (4, 1, 2)] variable
47 for shape in testcase_shapes:
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | xla_client_test.py | 2267 testcase_shapes = [
2311 } for dtype in dlpack_dtypes for shape in testcase_shapes
2390 for shape in testcase_shapes)
|