class Tensorflow::OpDef::ArgDef

Public Instance Methods

dtype() click to toggle source
# File lib/tensorflow/extensions/arg_def.rb, line 4
def dtype
  case self.type
    when :DT_INVALID
      nil
    when :DT_FLOAT
      :float
    when :DT_DOUBLE
      :double
    when :DT_INT32
      :int32
    when :DT_UINT8
      :uint8
    when :DT_INT16
      :int16
    when :DT_INT8
      :int8
    when :DT_STRING
      :string
    when :DT_COMPLEX64
      :complex64
    when :DT_INT64
      :int64
    when :DT_BOOL
      :bool
    when :DT_QINT8
      :qint8
    when :DT_QUINT8
      :quint8
    when :DT_QINT32
      :qint32
    when :DT_BFLOAT16
      :bfloat16
    when :DT_QINT16
      :qint16
    when :DT_QUINT16
      :quint16
    when :DT_UINT16
      :uint16
    when :DT_COMPLEX128
      :complex128
    when :DT_HALF
      :half
    when :DT_RESOURCE
      :resource
    when :DT_VARIANT
      :variant
    when :DT_UINT32
      :uint32
    when :DT_UINT64
      :uint64
  end
end