STenOptions

lamp.STenOptions
See theSTenOptions companion object
case class STenOptions(value: TensorOptions)

Attributes

Companion
object
Graph
Supertypes
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any
Show all

Members list

Value members

Concrete methods

def cpu[S : Sc]: STenOptions

Returns a copy with device set to CPU

Returns a copy with device set to CPU

Attributes

def cuda[S : Sc]: STenOptions

Returns a copy with device set to cuda:0

Returns a copy with device set to cuda:0

Attributes

def cudaIndex[S : Sc](index: Short): STenOptions

Returns a copy with device set to cuda with index

Returns a copy with device set to cuda with index

Attributes

def deviceIndex: Int
def isByte: Boolean
def isCPU: Boolean
def isCuda: Boolean
def isDouble: Boolean
def isFloat: Boolean
def isInt: Boolean
def isLong: Boolean
def isShort: Boolean
def isSparse: Boolean
def mps[S : Sc]: STenOptions

Returns a copy with device set to mps:0

Returns a copy with device set to mps:0

Attributes

def scalarTypeByte: Byte

Returns the byte representation of dtype

Returns the byte representation of dtype

0 - byte, 1 - char 2 - short 3 - int 4 - long, 5 - half 6 - float, 7 - double, 8 - complex half 9 - complex float 10 - complex double 11 - bool 15 - bfloat16 12,13,14,16 - quantized ints

Attributes

def toBF16[S : Sc]: STenOptions
def toByte[S : Sc]: STenOptions

Returns a copy with dtype set to a value compatible with Scala's Byte

Returns a copy with dtype set to a value compatible with Scala's Byte

Attributes

def toDouble[S : Sc]: STenOptions

Returns a copy with dtype set to double

Returns a copy with dtype set to double

Attributes

def toFloat[S : Sc]: STenOptions

Returns a copy with dtype set to float

Returns a copy with dtype set to float

Attributes

def toHalf[S : Sc]: STenOptions

Returns a copy with dtype set to half

Returns a copy with dtype set to half

Attributes

def toInt[S : Sc]: STenOptions

Returns a copy with dtype set to long

Returns a copy with dtype set to long

Attributes

def toLong[S : Sc]: STenOptions

Returns a copy with dtype set to long

Returns a copy with dtype set to long

Attributes

def toShort[S : Sc]: STenOptions

Returns a copy with dtype set to long

Returns a copy with dtype set to long

Attributes

Inherited methods

def productElementNames: Iterator[String]

Attributes

Inherited from:
Product
def productIterator: Iterator[Any]

Attributes

Inherited from:
Product