Searched defs:s32_arg_shape (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | shape_inference_test.cc | 1127 Shape s32_arg_shape = ShapeUtil::MakeShape(S32, {5, 3}); in TEST_F() local
1139 Shape s32_arg_shape = ShapeUtil::MakeShape(S32, {5, 3, 1}); in TEST_F() local
1165 Shape s32_arg_shape = ShapeUtil::MakeShape(S32, {5, 3}); in TEST_F() local
1178 Shape s32_arg_shape = ShapeUtil::MakeShape(S32, {5, 3}); in TEST_F() local
1201 Shape s32_arg_shape = ShapeUtil::MakeShape(S32, {5, 3, 1}); in TEST_F() local
1224 Shape s32_arg_shape = ShapeUtil::MakeShape(S32, {5, 3}); in TEST_F() local
1237 Shape s32_arg_shape = ShapeUtil::MakeShape(S32, {5, 3}); in TEST_F() local
1250 Shape s32_arg_shape = ShapeUtil::MakeShape(S32, {5, 3}); in TEST_F() local
|