MimIR
MimIR is my Intermediate Representation
Loading...
Searching...
No Matches
autodiff.h
Go to the documentation of this file.
1#pragma once
2
3#include <mim/world.h>
4
5#include "mim/plug/autodiff/autogen.h"
6
7namespace mim::plug::autodiff {
8
9/// @name autodiff.Tangent
10///@{
11const Def* tangent_type_fun(const Def*);
12///@}
13
14/// @name autodiff.zero
15///@{
16const Def* zero_def(const Def* T);
17const Def* zero_pullback(const Def* E, const Def* A);
18///@}
19
20/// @name autodiff.sum
21///@{
22const Def* op_sum(const Def* T, Defs);
23///@}
24
25/// @name Helpers
26///@{
27const Pi* pullback_type(const Def* E, const Def* A);
28const Def* id_pullback(const Def*);
29const Pi* autodiff_type_fun_pi(const Pi*);
30const Def* autodiff_type_fun(const Def*);
31///@}
32
33} // namespace mim::plug::autodiff
mim::Def
Base class for all Defs.
Definition def.h:273
mim::Pi
A dependent function type.
Definition lam.h:14
mim::plug::autodiff
The automatic differentiation Plugin
Definition autodiff.h:7
mim::plug::autodiff::autodiff_type_fun_pi
const Pi * autodiff_type_fun_pi(const Pi *)
Definition autodiff.cpp:83
mim::plug::autodiff::op_sum
const Def * op_sum(const Def *T, Defs)
Definition autodiff.cpp:151
mim::plug::autodiff::autodiff_type_fun
const Def * autodiff_type_fun(const Def *)
Definition autodiff.cpp:105
mim::plug::autodiff::zero_def
const Def * zero_def(const Def *T)
Definition autodiff.cpp:130
mim::plug::autodiff::tangent_type_fun
const Def * tangent_type_fun(const Def *)
Definition autodiff.cpp:54
mim::plug::autodiff::zero_pullback
const Def * zero_pullback(const Def *E, const Def *A)
Definition autodiff.cpp:42
mim::plug::autodiff::id_pullback
const Def * id_pullback(const Def *)
Definition autodiff.cpp:30
mim::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:59
mim::Defs
fe::View< const Def * > Defs
Definition def.h:91