grain ~master (2021-05-15T05:53:16Z)

to

grain autograd

CPU -> CUDA memory conversion

  1. auto to(T[] src)
  2. auto to(Src src)
    version(grain_cuda)
    auto
    to
    (
    alias S : HostStorage
    Src
    )
    (
    Src src
    )
    if (
    isDevice!Src
    )
  3. Variable!(T, dim, Dst) to(Variable!(T, dim, Src) src)

Examples

auto h = [[0.1f, 0.2f, 0.3f], [0.4f, 0.5f, 0.6f]].variable;
auto d = h.to!DeviceStorage;
assert(h.data == d.to!HostStorage.data);

Meta

Source

See Implementation

grain autograd aliasesfunctionsstructsvariables

Page generated by adrdox