layernorm_node.h File Reference

#include <cstddef>
#include <memory>
#include <optional>
#include <string>
#include <nlohmann/json_fwd.hpp>
#include "sdfg/builder/structured_sdfg_builder.h"
#include "sdfg/data_flow/data_flow_graph.h"
#include "sdfg/data_flow/data_flow_node.h"
#include "sdfg/data_flow/library_node.h"
#include "sdfg/data_flow/library_nodes/math/tensor/tensor_layout.h"
#include "sdfg/data_flow/library_nodes/math/tensor/tensor_node.h"
#include "sdfg/data_flow/pointer_metadata.h"
#include "sdfg/graph/graph.h"
#include "sdfg/serializer/json_serializer.h"
#include "sdfg/structured_control_flow/block.h"
#include "sdfg/symbolic/symbolic.h"

Go to the source code of this file.

Classes
class	sdfg::math::tensor::LayerNormNode
	Applies layer normalization over the trailing noramlized_shape dimensions of the input. More...
class	sdfg::math::tensor::LayerNormNodeSerializer

Namespaces
namespace	sdfg
namespace	sdfg::math
namespace	sdfg::math::tensor

Functions
data_flow::LibraryNodeCode	sdfg::math::tensor::LibraryNodeType_LayerNorm ("ml::LayerNorm")