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

* Adding cast to fp32

* Adding myself to OWNERS

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

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