Adding cast to fp32 for TensorRT no image case (tensorflow#3917)

karmel · web-flow · commit 18e064386b3a · 2018-04-09T10:09:43.000-07:00
* Adding cast to fp32

* Adding myself to OWNERS
diff --git a/CODEOWNERS b/CODEOWNERS
@@ -39,6 +39,7 @@
 /research/swivel/ @waterson
 /research/syntaxnet/ @calberti @andorardo @bogatyy @markomernick
 /research/tcn/ @coreylynch @sermanet
+/research/tensorrt/ @karmel
 /research/textsum/ @panyx0718 @peterjliu
 /research/transformer/ @daviddao
 /research/video_prediction/ @cbfinn
diff --git a/research/tensorrt/tensorrt.py b/research/tensorrt/tensorrt.py
@@ -118,7 +118,8 @@ def batch_from_random(batch_size, output_height=224, output_width=224,
       [batch_size, output_height, output_width, num_channels]
   """
   shape = [batch_size, output_height, output_width, num_channels]
-  return np.random.random_sample(shape)
+  # Make sure we return float32, as float64 will not get cast automatically.
+  return np.random.random_sample(shape).astype(np.float32)
 
 
 ################################################################################
