tiny.accy.tensor

1pub const tensor = @import("tensor/root.zig");

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