Searched refs:zx_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder_test.cc | 109 auto zx_shape = ShapeUtil::MakeShape(F32, {zdim, xdim}); in TEST_F() local
129 zx_shape, a_param, b_param, /*feature_group_count=*/1, in TEST_F()
152 auto zx_shape = ShapeUtil::MakeShape(F32, {zdim, xdim}); in TEST_F() local
172 zx_shape, a_param, b_param, /*feature_group_count=*/1, in TEST_F()
|
| D | dynamic_dimension_inference_test.cc | 293 auto zx_shape = ShapeUtil::MakeShape(F32, {zdim, xdim}); in TEST_F() local
313 zx_shape, a_param, b_param, /*feature_group_count=*/1, in TEST_F()
|