elu_node.h File Reference

ELU tensor node with PyTorch semantics. More...

#include <memory>
#include <nlohmann/json_fwd.hpp>
#include <string>
#include <vector>
#include "sdfg/builder/structured_sdfg_builder.h"
#include "sdfg/data_flow/library_nodes/math/tensor/tensor_node.h"
#include "sdfg/function.h"
#include "sdfg/passes/expansion/lib_node_expander.h"
#include "sdfg/serializer/json_serializer.h"
#include "sdfg/symbolic/symbolic.h"

Go to the source code of this file.

Classes
class	sdfg::math::tensor::EluNode
	Computes Y = X < 0 ? More...
class	sdfg::math::tensor::EluNodeSerializer

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

Functions
data_flow::LibraryNodeCode	sdfg::math::tensor::LibraryNodeType_Elu ("ml::Elu")

Detailed Description

ELU tensor node with PyTorch semantics.