Update TensorIterator-inl.h