Python infeed_dequeue Examples

Example #1

    def infeed_dequeue(dtype, shape, name=None):
        """A placeholder op for a value that will be fed into the computation.

    Args:
      dtype: A `tf.DType`. The type of elements in the tensor.
      shape: A `tf.TensorShape` or list of `ints`. The shape of the tensor.
      name: A name for the operation (optional).

    Returns:
      A `Tensor` of type `dtype`.
      A tensor that will be provided using the infeed mechanism.

    Raises:
      TypeError: If 'dtype` is not a supported infeed type.
    """
        if dtype not in _SUPPORTED_INFEED_DTYPES:
            raise TypeError(
                "{} is not a supported TPU infeed type. Supported types are: "
                "{}".format(dtype, list(_SUPPORTED_INFEED_DTYPES)))

        return gen_tpu_ops.infeed_dequeue(dtype, shape, name=name)

Example #2

File: tpu_ops.py Project: adit-chandra/tensorflow

def infeed_dequeue(dtype, shape, name=None):
  """A placeholder op for a value that will be fed into the computation.

  Args:
    dtype: A `tf.DType`. The type of elements in the tensor.
    shape: A `tf.TensorShape` or list of `ints`. The shape of the tensor.
    name: A name for the operation (optional).

  Returns:
    A `Tensor` of type `dtype`.
    A tensor that will be provided using the infeed mechanism.

  Raises:
    TypeError: If 'dtype` is not a supported infeed type.
  """
  if dtype not in _SUPPORTED_INFEED_DTYPES:
    raise TypeError(
        "{} is not a supported TPU infeed type. Supported types are: "
        "{}".format(dtype, list(_SUPPORTED_INFEED_DTYPES)))

  return gen_tpu_ops.infeed_dequeue(dtype, shape, name=name)