message: Fixes a bug in the quantize_and_dequantize_op kernel with getting the min and max range,
  when the op is on the GPU but the range tensor is on the host.
diff: @@ -47,9 +47,13 @@ struct QuantizeAndDequantizeOneScaleImpl {
  if (!range_given) {
  input_min.device(d) = input.minimum();
  input_max.device(d) = input.maximum();
  - }
  d.memcpyDeviceToHost(&min_range, input_min.data(), sizeof(T));
  d.memcpyDeviceToHost(&max_range, input_max.data(), sizeof(T));
  + } else {
  + // Copy the range values from their respective tensors on the host.
  + min_range = input_min_tensor->scalar<T>()();
  + max_range = input_max_tensor->scalar<T>()();
  + }
  // Calculate the range for the simulated integer quantization:
  // e.g. [-128,127] for signed = true, num_bits = 8,
  