leaky_relu_node.h File Reference

Leaky ReLU 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::LeakyReLUNode
	Computes Y = X > 0 ? More...
class	sdfg::math::tensor::LeakyReLUNodeSerializer

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

Functions
data_flow::LibraryNodeCode	sdfg::math::tensor::LibraryNodeType_LeakyReLU ("ml::LeakyReLU")

Detailed Description

Leaky ReLU tensor node with PyTorch semantics.