Searched refs:transpose_y (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transpose_folding_test.cc | 227 HloInstruction* transpose_y = in TEST_F() local
240 transpose_y->shape().dimensions(dnums.kernel_spatial_dimensions(i))); in TEST_F()
243 x->shape(), transpose_y->shape(), /*feature_group_count=*/1, in TEST_F()
247 conv_shape.ValueOrDie(), x, transpose_y, in TEST_F()
283 HloInstruction* transpose_y = in TEST_F() local
296 transpose_y->shape().dimensions(dnums.kernel_spatial_dimensions(i))); in TEST_F()
299 x->shape(), transpose_y->shape(), /*feature_group_count=*/1, in TEST_F()
303 conv_shape.ValueOrDie(), x, transpose_y, in TEST_F()
|