embedding_renorm_node.h File Reference

#include "sdfg/data_flow/library_nodes/math/tensor/tensor_layout.h"
#include "sdfg/data_flow/library_nodes/math/tensor/tensor_node.h"
#include "sdfg/serializer/json_serializer.h"

Go to the source code of this file.

Classes
class	sdfg::math::tensor::EmbeddingRenormNode
	In-place embedding renormalization implementing aten.embedding_renorm_. More...
class	sdfg::math::tensor::EmbeddingRenormNodeSerializer

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

Functions
data_flow::LibraryNodeCode	sdfg::math::tensor::LibraryNodeType_EmbeddingRenorm ("ml::EmbeddingRenorm")