| 435 | } |
| 436 | |
| 437 | void TensorShapeOld::RemoveDim(int d) { |
| 438 | CHECK_GE(d, 0); |
| 439 | CHECK_LT(d, dims()); |
| 440 | |
| 441 | // Update the number of elements and remove the dimension from the |
| 442 | // sizes. |
| 443 | dim_sizes_.erase(dim_sizes_.begin() + d); |
| 444 | recompute_dims(); |
| 445 | } |
| 446 | |
| 447 | void TensorShapeOld::recompute_dims() { |
| 448 | num_elements_ = 1; |
no test coverage detected