clamp_node.h File Reference

Clamp 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::ClampNode
	Computes Y = min(max(X, min), max) elementwise, where either bound may be omitted. More...
class	sdfg::math::tensor::ClampNodeSerializer

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

Functions
data_flow::LibraryNodeCode	sdfg::math::tensor::LibraryNodeType_Clamp ("ml::Clamp")

Detailed Description

Clamp tensor node with PyTorch semantics.