thorin2/autodiff_8h_source.html

#pragma once


#include "thorin/world.h"


#include "thorin/plug/autodiff/autogen.h"


namespace thorin::plug::autodiff {


/// @name %%autodiff.Tangent

///@{

const Def* tangent_type_fun(const Def*);

///@}


/// @name %%autodiff.zero

///@{

const Def* zero_def(const Def* T);

const Def* zero_pullback(const Def* E, const Def* A);

///@}


/// @name %%autodiff.sum

///@{

const Def* op_sum(const Def* T, Defs);

///@}


/// @name Helpers

///@{

const Pi* pullback_type(const Def* E, const Def* A);

const Def* id_pullback(const Def*);

const Pi* autodiff_type_fun_pi(const Pi*);

const Def* autodiff_type_fun(const Def*);

///@}


} // namespace thorin::plug::autodiff


autogen.h

thorin::Def
Base class for all Defs.
Definition def.h:220

thorin::Pi
A dependent function type.
Definition lam.h:11

thorin::Span
This is a thin wrapper for std::span<T, N> with the following additional features:
Definition span.h:28

thorin::plug::autodiff
The automatic differentiation Plugin
Definition autodiff.h:7

thorin::plug::autodiff::tangent_type_fun
const Def * tangent_type_fun(const Def *)
Definition autodiff.cpp:57

thorin::plug::autodiff::zero_def
const Def * zero_def(const Def *T)
Definition autodiff.cpp:140

thorin::plug::autodiff::pullback_type
const Pi * pullback_type(const Def *E, const Def *A)
computes pb type E* -> A* E - type of the expression (return type for a function) A - type of the arg...
Definition autodiff.cpp:62

thorin::plug::autodiff::autodiff_type_fun_pi
const Pi * autodiff_type_fun_pi(const Pi *)
Definition autodiff.cpp:90

thorin::plug::autodiff::id_pullback
const Def * id_pullback(const Def *)
Definition autodiff.cpp:32

thorin::plug::autodiff::zero_pullback
const Def * zero_pullback(const Def *E, const Def *A)
Definition autodiff.cpp:44

thorin::plug::autodiff::op_sum
const Def * op_sum(const Def *T, Defs)
Definition autodiff.cpp:169

thorin::plug::autodiff::autodiff_type_fun
const Def * autodiff_type_fun(const Def *)
Definition autodiff.cpp:113

world.h