docc 0.8.3 (0d44999b)
C++ API Reference
Loading...
Searching...
No Matches
gelu_node.h File Reference
#include <cstddef>
#include <memory>
#include <string>
#include <vector>
#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/elementwise_node.h"
#include "sdfg/data_flow/library_nodes/math/tensor/tensor_node.h"
#include "sdfg/element.h"
#include "sdfg/graph/graph.h"
#include "sdfg/structured_control_flow/block.h"
#include "sdfg/symbolic/symbolic.h"
#include "sdfg/types/type.h"

Go to the source code of this file.

Classes

class  sdfg::math::tensor::GELUNode
 Applies the Gaussion Error Linear Units function. More...
class  sdfg::math::tensor::GELUNodeSerializer

Namespaces

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

Functions

data_flow::LibraryNodeCode sdfg::math::tensor::LibraryNodeType_GELU ("ml::GELU")