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

* Adding cast to fp32 * Adding myself to OWNERS

Adding cast to fp32 for TensorRT no image case (#3917)
* Adding cast to fp32 * Adding myself to OWNERS
18e06438 · Karmel Allison · GitHub · b6bcc450 · 18e06438 · 18e06438
Unverified Commit 18e06438 authored Apr 09, 2018 by Karmel Allison Committed by GitHub Apr 09, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

CODEOWNERS CODEOWNERS +1 -0

research/tensorrt/tensorrt.py research/tensorrt/tensorrt.py +2 -1

No files found.
--- 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

--- 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)
 ################################################################################