neon.data.dataloader_transformers.TypeCast

class neon.data.dataloader_transformers.TypeCast(dataloader, index, dtype, *args, **kwargs)[source]

Bases: neon.data.dataloader_transformers.DataLoaderTransformer

TypeCast data from dataloader at index to dtype and move into device memory if not already.

__init__(dataloader, index, dtype, *args, **kwargs)[source]

Methods

__init__(dataloader, index, dtype, *args, …)
gen_class(pdict)
get_description([skip]) Returns a dict that contains all necessary information needed to serialize this object.
recursive_gen(pdict, key) helper method to check whether the definition
transform(t)
be = None
classnm

Returns the class name.

gen_class(pdict)
get_description(skip=[], **kwargs)

Returns a dict that contains all necessary information needed to serialize this object.

Parameters:skip (list) – Objects to omit from the dictionary.
Returns:Dictionary format for object information.
Return type:(dict)
modulenm

Returns the full module path.

recursive_gen(pdict, key)

helper method to check whether the definition dictionary is defining a NervanaObject child, if so it will instantiate that object and replace the dictionary element with an instance of that object

transform(t)[source]