From 4de7361c43adc65c442359104a17c449a02eee7a Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Thu, 6 Jul 2017 10:01:25 -0700
Subject: [PATCH] Fix some internal tests.

PiperOrigin-RevId: 161093488
---
 .../contrib/tensor_forest/kernels/v4/input_target.h      | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/tensorflow/contrib/tensor_forest/kernels/v4/input_target.h b/tensorflow/contrib/tensor_forest/kernels/v4/input_target.h
index 03dc332c8ad..0309ec1de9a 100644
--- a/tensorflow/contrib/tensor_forest/kernels/v4/input_target.h
+++ b/tensorflow/contrib/tensor_forest/kernels/v4/input_target.h
@@ -20,7 +20,7 @@
 namespace tensorflow {
 namespace tensorforest {
 
-typedef TTypes<float, 1>::ConstTensor SingleDimStorageType;
+typedef TTypes<float, 1>::UnalignedConstTensor SingleDimStorageType;
 
 // Base class for classes that hold labels and weights. Mostly for testing
 // purposes, because it's inconvenient to construct nasty Eigen::things.
@@ -54,9 +54,10 @@ class StoredInputTarget : public InputTarget {
 class TensorInputTarget : public StoredInputTarget<SingleDimStorageType> {
  public:
   TensorInputTarget(const Tensor& target, const Tensor& weight, int num_targets)
-      : StoredInputTarget(new SingleDimStorageType(target.tensor<float, 1>()),
-                          new SingleDimStorageType(weight.tensor<float, 1>()),
-                          num_targets),
+      : StoredInputTarget(
+            new SingleDimStorageType(target.unaligned_flat<float>()),
+            new SingleDimStorageType(weight.unaligned_flat<float>()),
+            num_targets),
         original_tensor_(target) {}
 
   int32 GetTargetAsClassIndex(int example_index,