tensor_node.h File Reference

Abstract base class for tensor library nodes. More...

#include "sdfg/data_flow/library_nodes/math/math_node.h"
#include "sdfg/types/type.h"

Go to the source code of this file.

Classes
struct	sdfg::math::tensor::ContractionTypes
	Element types of a contraction such as a matmul. More...
class	sdfg::math::tensor::TensorNode
	Abstract base class for all tensor operations. More...

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

Typedefs
typedef types::PrimitiveType	sdfg::math::tensor::QuantizationType

Functions
types::PrimitiveType	sdfg::math::tensor::deserialize_quantization (const nlohmann::json &j, const std::string &field_name, types::PrimitiveType default_value=QUANTIZATION_MATCH_INPUTS)

Variables
constexpr QuantizationType	sdfg::math::tensor::QUANTIZATION_MATCH_INPUTS = types::PrimitiveType::Void

Detailed Description

Abstract base class for tensor library nodes.

This file defines the abstract base class TensorNode which provides common validation and type determination functionality for all tensor operations.

Type Requirements

Tensor nodes require that:

• All memlets are scalar or pointer of scalar types
• All connected memlets have the same primitive type (e.g., all Float or all Int32)
• The primitive type is consistent across inputs and outputs