Searched refs:yz_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder_test.cc | 108 auto yz_shape = ShapeUtil::MakeShape(F32, {ydim, zdim}); in TEST_F() local
114 /*parameter_number=*/1, yz_shape, "B")); in TEST_F()
151 auto yz_shape = ShapeUtil::MakeShape(F32, {ydim, zdim}); in TEST_F() local
157 /*parameter_number=*/1, yz_shape, "B")); in TEST_F()
|
| D | dynamic_dimension_inference_test.cc | 248 auto yz_shape = ShapeUtil::MakeShape(F32, {ydim, zdim}); in TEST_F() local
254 /*parameter_number=*/1, yz_shape, "B")); in TEST_F()
292 auto yz_shape = ShapeUtil::MakeShape(F32, {ydim, zdim}); in TEST_F() local
298 /*parameter_number=*/1, yz_shape, "B")); in TEST_F()
|