lib/accy/src/tensor/root.zig

daab053ee43316e1809a84551d573ddd1e5bf3d2

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