MimIR
MimIR is my Intermediate Representation
Loading...
Searching...
No Matches
mim::plug::btensor::MapRedPost Struct Reference

The curried meta arguments of a tensor.map_reduce_post or btensor.map_reduce_post callee (same layout). More...

#include <mim/plug/btensor/btensor.h>

Classes

struct  Ranks

Public Member Functions

std::optional< Ranks > ranks () const
 The literal input counts and loop ranks; std::nullopt unless all are literal and Rn >= Ro.

Static Public Member Functions

static MapRedPost of (const App *callee)

Public Attributes

const Def * nis_nps
const Def * meta
const Def * shapes
const Def * in_tys
const Def * comb_init
const Def * map_out
const Def * maps_all
const Def * nis
const Def * nps
const Def * To
const Def * Tp
const Def * Ro
const Def * Rn
const Def * TSched
const Def * So
const Def * Sr
const Def * sched
const Def * Tis
const Def * Ris
const Def * Sis
const Def * Tps
const Def * Rps
const Def * Sps
const Def * comb
const Def * init
const Def * post
const Def * maps
const Def * post_maps

Detailed Description

The curried meta arguments of a tensor.map_reduce_post or btensor.map_reduce_post callee (same layout).

Definition at line 20 of file btensor.h.

Member Function Documentation

◆ of()

MapRedPost mim::plug::btensor::MapRedPost::of ( const App * callee)
inlinestatic

Definition at line 25 of file btensor.h.

References comb, comb_init, in_tys, init, map_out, maps, maps_all, meta, nis, nis_nps, nps, post, post_maps, mim::Def::projs(), Ris, Rn, Ro, Rps, sched, shapes, Sis, So, Sps, Sr, Tis, To, Tp, Tps, TSched, and mim::App::uncurry_args().

◆ ranks()

std::optional< Ranks > mim::plug::btensor::MapRedPost::ranks ( ) const
inline

The literal input counts and loop ranks; std::nullopt unless all are literal and Rn >= Ro.

Definition at line 44 of file btensor.h.

References mim::Lit::isa(), nis, nps, Rn, and Ro.

Member Data Documentation

◆ comb

const Def * mim::plug::btensor::MapRedPost::comb

Definition at line 23 of file btensor.h.

Referenced by of().

◆ comb_init

const Def * mim::plug::btensor::MapRedPost::comb_init

Definition at line 21 of file btensor.h.

Referenced by of().

◆ in_tys

const Def * mim::plug::btensor::MapRedPost::in_tys

Definition at line 21 of file btensor.h.

Referenced by of().

◆ init

const Def * mim::plug::btensor::MapRedPost::init

Definition at line 23 of file btensor.h.

Referenced by of().

◆ map_out

const Def * mim::plug::btensor::MapRedPost::map_out

Definition at line 21 of file btensor.h.

Referenced by of().

◆ maps

const Def * mim::plug::btensor::MapRedPost::maps

Definition at line 23 of file btensor.h.

Referenced by of().

◆ maps_all

const Def * mim::plug::btensor::MapRedPost::maps_all

Definition at line 21 of file btensor.h.

Referenced by of().

◆ meta

const Def * mim::plug::btensor::MapRedPost::meta

Definition at line 21 of file btensor.h.

Referenced by of().

◆ nis

const Def* mim::plug::btensor::MapRedPost::nis

Definition at line 22 of file btensor.h.

Referenced by of(), and ranks().

◆ nis_nps

const Def* mim::plug::btensor::MapRedPost::nis_nps

Definition at line 21 of file btensor.h.

Referenced by of().

◆ nps

const Def * mim::plug::btensor::MapRedPost::nps

Definition at line 22 of file btensor.h.

Referenced by of(), and ranks().

◆ post

const Def * mim::plug::btensor::MapRedPost::post

Definition at line 23 of file btensor.h.

Referenced by of().

◆ post_maps

const Def * mim::plug::btensor::MapRedPost::post_maps

Definition at line 23 of file btensor.h.

Referenced by of().

◆ Ris

const Def * mim::plug::btensor::MapRedPost::Ris

Definition at line 23 of file btensor.h.

Referenced by of().

◆ Rn

const Def * mim::plug::btensor::MapRedPost::Rn

Definition at line 22 of file btensor.h.

Referenced by of(), and ranks().

◆ Ro

const Def * mim::plug::btensor::MapRedPost::Ro

Definition at line 22 of file btensor.h.

Referenced by of(), and ranks().

◆ Rps

const Def * mim::plug::btensor::MapRedPost::Rps

Definition at line 23 of file btensor.h.

Referenced by of().

◆ sched

const Def * mim::plug::btensor::MapRedPost::sched

Definition at line 22 of file btensor.h.

Referenced by of().

◆ shapes

const Def * mim::plug::btensor::MapRedPost::shapes

Definition at line 21 of file btensor.h.

Referenced by of().

◆ Sis

const Def * mim::plug::btensor::MapRedPost::Sis

Definition at line 23 of file btensor.h.

Referenced by of().

◆ So

const Def * mim::plug::btensor::MapRedPost::So

Definition at line 22 of file btensor.h.

Referenced by of().

◆ Sps

const Def * mim::plug::btensor::MapRedPost::Sps

Definition at line 23 of file btensor.h.

Referenced by of().

◆ Sr

const Def * mim::plug::btensor::MapRedPost::Sr

Definition at line 22 of file btensor.h.

Referenced by of().

◆ Tis

const Def* mim::plug::btensor::MapRedPost::Tis

Definition at line 23 of file btensor.h.

Referenced by of().

◆ To

const Def * mim::plug::btensor::MapRedPost::To

Definition at line 22 of file btensor.h.

Referenced by of().

◆ Tp

const Def * mim::plug::btensor::MapRedPost::Tp

Definition at line 22 of file btensor.h.

Referenced by of().

◆ Tps

const Def * mim::plug::btensor::MapRedPost::Tps

Definition at line 23 of file btensor.h.

Referenced by of().

◆ TSched

const Def * mim::plug::btensor::MapRedPost::TSched

Definition at line 22 of file btensor.h.

Referenced by of().

The documentation for this struct was generated from the following file: