message: Remove reference to forwarding header.
diff: @@ -13,11 +13,12 @@ See the License for the specific language governing permissions and
  limitations under the License.
  ==============================================================================*/
  #include <queue>
  +
  +#include "tensorflow/core/framework/dataset.h"
  #include "tensorflow/core/framework/op_kernel.h"
  #include "tensorflow/core/framework/partial_tensor_shape.h"
  #include "tensorflow/core/framework/tensor.h"
  #include "tensorflow/core/framework/tensor_shape.h"
  -#include "tensorflow/core/kernels/data/dataset.h"
  #include "tensorflow/core/lib/core/blocking_counter.h"
  #include "tensorflow/core/lib/core/errors.h"
  #include "tensorflow/core/lib/core/threadpool.h"
  