normalizers.cpp File Reference

#include <algorithm>
#include <array>
#include <fe/format.h>
#include <mim/def.h>
#include <mim/driver.h>
#include <mim/plugin.h>
#include <mim/tuple.h>
#include <mim/world.h>
#include <mim/plug/affine/affine.h>
#include "mim/plug/tensor/tensor.h"

Include dependency graph for normalizers.cpp:

Go to the source code of this file.

Namespaces
namespace	mim
namespace	mim::plug
namespace	mim::plug::tensor
	The tensor Plugin

Functions
<tt>tensor.broadcast</tt>
const Def *	mim::plug::tensor::normalize_broadcast (const Def *type, const Def *c, const Def *arg)
<tt>tensor.broadcast_in_dim</tt>
const Def *	mim::plug::tensor::normalize_broadcast_in_dim (const Def *, const Def *, const Def *)
<tt>tensor.repeat</tt>
const Def *	mim::plug::tensor::normalize_repeat (const Def *type, const Def *c, const Def *arg)
<tt>tensor.reshape</tt>
const Def *	mim::plug::tensor::normalize_reshape (const Def *type, const Def *c, const Def *arg)
<tt>tensor.transpose</tt>
const Def *	mim::plug::tensor::normalize_transpose (const Def *, const Def *c, const Def *arg)
<tt>tensor.slice</tt>
const Def *	mim::plug::tensor::normalize_slice (const Def *, const Def *c, const Def *arg)
<tt>tensor.flip</tt>
const Def *	mim::plug::tensor::normalize_flip (const Def *, const Def *, const Def *)
<tt>tensor.pad</tt>
const Def *	mim::plug::tensor::normalize_pad (const Def *, const Def *c, const Def *arg)
<tt>tensor.concat</tt>
const Def *	mim::plug::tensor::normalize_concat (const Def *, const Def *, const Def *)
<tt>tensor.if_static</tt>
const Def *	mim::plug::tensor::normalize_if_static (const Def *, const Def *, const Def *arg)
<tt>tensor.fastest_axis</tt>
const Def *	mim::plug::tensor::normalize_fastest_axis (const Def *, const Def *, const Def *arg)
<tt>tensor.vec_width</tt>
const Def *	mim::plug::tensor::normalize_vec_width (const Def *, const Def *, const Def *arg)
<tt>tensor.shape</tt>
const Def *	mim::plug::tensor::normalize_shape (const Def *, const Def *c, const Def *arg)

Variables
constexpr std::array	mim::plug::tensor::Vec_widths = {1_u64, 4_u64, 8_u64, 16_u64}
	The vector widths -X tensor:vec-width accepts.