Python copy_tensor_or_indexed_slices_to_device示例

示例#1

文件： cross_tower_ops.py 项目： AnishShah/tensorflow

def _simple_broadcast(value, destinations):
  index = {}
  devices = get_devices_from(destinations)
  for d in devices:
    index[d] = cross_tower_utils.copy_tensor_or_indexed_slices_to_device(
        value, d)
  return value_lib.Mirrored(index)

示例#2

def _simple_broadcast(value, destinations):
  index = {}
  devices = get_devices_from(destinations)
  for d in devices:
    index[d] = cross_tower_utils.copy_tensor_or_indexed_slices_to_device(
        value, d)
  return value_lib.Mirrored(index)

示例#3

文件： cross_tower_utils_test.py 项目： abhinav-upadhyay/tensorflow

  def testCopyTensor(self):
    with ops.device("/cpu:0"):
      t = constant_op.constant([[1., 2.], [0, 0], [3., 4.]])
    destination = "/gpu:0"
    result = cross_tower_utils.copy_tensor_or_indexed_slices_to_device(
        t, destination)

    self._assert_values_equal(t, result)
    self.assertEqual(device_util.resolve(destination),
                     device_util.resolve(result.device))

示例#4

文件： cross_tower_utils_test.py 项目： zhikun0704/tensorflow

    def testCopyTensor(self):
        with ops.device("/cpu:0"):
            t = constant_op.constant([[1., 2.], [0, 0], [3., 4.]])
        destination = "/gpu:0"
        result = cross_tower_utils.copy_tensor_or_indexed_slices_to_device(
            t, destination)

        self._assert_values_equal(t, result)
        self.assertEqual(device_util.resolve(destination),
                         device_util.resolve(result.device))