Searched refs:warp_np (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/resampler/xla/ |
| D | resampler_ops_xla_test.py | 32 def _assertForwardOpMatchesExpected(self, image_np, warp_np, expected): argument
35 warp = array_ops.placeholder(warp_np.dtype)
37 out = sess.run(resampled, {input_image: image_np, warp: warp_np})
42 def _assertBackwardOpMatchesExpected(self, input_np, warp_np, grad_output_np, argument
46 warp = array_ops.placeholder(warp_np.dtype)
54 warp: warp_np,
71 warp_np = np.array(warp_data, dtype=dtype).reshape(warp_shape)
73 self._assertForwardOpMatchesExpected(input_np, warp_np, expected)
82 self._assertBackwardOpMatchesExpected(input_np, warp_np, grad_output,
94 warp_np = np.array(warp_data, dtype=dtype).reshape(warp_shape)
[all …]
|