Searched refs:use_reshape (Results 1 – 2 of 2) sorted by relevance
1267 bool use_reshape = (x.dims() == 5); in ComputeWithReservedSpace() local1270 if (use_reshape) { in ComputeWithReservedSpace()1300 auto alloc_shape = use_reshape ? dest_shape : x_shape; in ComputeWithReservedSpace()1330 if (use_reshape) { in ComputeWithReservedSpace()1428 bool use_reshape = (x.dims() == 5); in ComputeWithReservedSpace() local1431 if (use_reshape) { in ComputeWithReservedSpace()1446 auto alloc_shape = use_reshape ? dest_shape : x_shape; in ComputeWithReservedSpace()1492 if (use_reshape) { in ComputeWithReservedSpace()
3198 for (bool use_reshape : {true, false}) { in TEST_F()3207 ops::Placeholder::Shape(PartialTensorShape({use_reshape ? -1 : 2}))); in TEST_F()3209 if (use_reshape) { in TEST_F()3216 item.fetch.push_back(use_reshape ? "reshape" : "sum"); in TEST_F()