Repository navigation
Expand file tree
/
Copy pathroot.zig
More file actions
96 lines (95 loc) · 4.34 KB
/
Copy pathroot.zig
File metadata and controls
96 lines (95 loc) · 4.34 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
pub const types = @import("type/root.zig");
pub const program = @import("program.zig");
pub const dsl = @import("dsl/root.zig");
pub const interpret = @import("interpret/root.zig");
pub const random = @import("random/root.zig");
pub const trace = @import("trace/root.zig");
pub const transform = @import("transform.zig");
pub const autodiff = @import("autodiff.zig");
pub const batch = @import("batch.zig");
pub const reverse = @import("reverse.zig");
pub const gradient = @import("grad.zig");
pub const nn = @import("nn.zig");
pub const lower = @import("lower.zig");
pub const execute = @import("execute.zig");
pub const function = @import("function.zig");
pub const unroll = @import("unroll.zig");
pub const wire = @import("wire/root.zig");
pub const session = @import("session/root.zig");
pub const DType = types.DType;
pub const Type = types.Type;
pub const Spec = types.Spec;
pub const Dim = types.Dim;
pub const Graph = program.Program;
pub const Program = dsl.Program;
pub const program_product_name = program.product_name;
pub const Id = program.Id;
pub const Operation = program.Operation;
pub const Unary = program.Unary;
pub const Binary = program.Binary;
pub const Reducer = program.Reducer;
pub const Builder = trace.Builder;
pub const Value = trace.Value;
pub const Dual = autodiff.Dual;
pub const TransformContext = transform.Context;
pub const spec = types.spec;
pub const specDims = types.specDims;
pub const define = trace.define;
pub const rewrite = transform.apply;
pub const linearize = autodiff.linearize;
pub const linearizeWith = autodiff.linearizeWith;
pub const jvp = autodiff.jvp;
pub const jvpWith = autodiff.jvpWith;
pub const pullback = reverse.pullback;
pub const pullbackWith = reverse.pullbackWith;
pub const grad = gradient.grad;
pub const valueAndGrad = gradient.valueAndGrad;
pub const gradWith = gradient.gradWith;
pub const gradWithRules = gradient.gradWithRules;
pub const embedding = nn.embedding;
pub const embeddingNamed = nn.embeddingNamed;
pub const sparseCrossEntropy = nn.sparseCrossEntropy;
pub const sparseCrossEntropyNamed = nn.sparseCrossEntropyNamed;
pub const sparseCrossEntropyMean = nn.sparseCrossEntropyMean;
pub const sparseCrossEntropyMeanNamed = nn.sparseCrossEntropyMeanNamed;
pub const vmap = batch.vmap;
pub const vmapWith = batch.vmapWith;
pub const LinearizeOptions = autodiff.LinearizeOptions;
pub const Linearization = autodiff.Linearization;
pub const Pullback = reverse.Pullback;
pub const PullbackOptions = reverse.Options;
pub const GradOptions = gradient.Options;
pub const JvpOptions = autodiff.JvpOptions;
pub const VmapOptions = batch.Options;
pub const BatchAxis = batch.Axis;
pub const mappedAxis = batch.mapped;
pub const runCpu = execute.runCpu;
pub const CpuExecutor = execute.Cpu;
pub const Function = function.Function;
pub const toSemanticModule = lower.toSemanticModule;
pub const prepare = lower.prepare;
pub const prepareWith = lower.prepareWith;
pub const prepareFragment = lower.prepareFragment;
pub const createArtifactJob = lower.createArtifactJob;
pub const createArtifactJobFromPreparedJob = lower.createArtifactJobFromPreparedJob;
pub const compileFragmentFromArtifactJob = lower.compileFragmentFromArtifactJob;
pub const compileFragment = lower.compileFragment;
pub const compileFragmentFromPreparedJob = lower.compileFragmentFromPreparedJob;
pub const BackendPreparedJob = lower.BackendPreparedJob;
pub const BackendPreparationRunOptions = lower.BackendPreparationRunOptions;
pub const GeneratedScheduleKind = lower.GeneratedScheduleKind;
pub const GeneratedSchedule = lower.GeneratedSchedule;
pub const GeneratedKernelProgram = lower.GeneratedKernelProgram;
pub const GeneratedKernelSummary = lower.GeneratedKernelSummary;
pub const GeneratedKernelSummaries = lower.GeneratedKernelSummaries;
pub const ArtifactJob = lower.ArtifactJob;
pub const BackendHandle = lower.BackendHandle;
pub const CompiledFragment = lower.CompiledFragment;
pub const LoadedFragment = lower.LoadedFragment;
pub const FragmentCompilerOptions = lower.FragmentCompilerOptions;
pub const FragmentCompilerCache = lower.FragmentCompilerCache;
pub const FragmentCompilerCacheUpdate = lower.FragmentCompilerCacheUpdate;
pub const FragmentCompilationRequest = lower.FragmentCompilationRequest;
pub const ArtifactKernelSource = lower.ArtifactKernelSource;
pub const ArtifactKernelSummary = lower.ArtifactKernelSummary;
pub const ArtifactKernelSummaries = lower.ArtifactKernelSummaries;