fill_node.h

Go to the documentation of this file.

#pragma once
 
#include "sdfg/data_flow/library_nodes/math/tensor/elementwise_node.h"
 
namespace sdfg {
namespace math {
namespace tensor {
 
inline data_flow::LibraryNodeCode LibraryNodeType_Fill("ml::Fill");
 
class FillNode : public ElementWiseDataflowTensorNode {
public:
    FillNode(
        size_t element_id,
        const DebugInfo& debug_info,
        const graph::Vertex vertex,
        data_flow::DataFlowGraph& parent,
        const std::vector<symbolic::Expression>& shape,
        QuantizationType quantization = QUANTIZATION_MATCH_INPUTS,
        const data_flow::ImplementationType& impl_type = data_flow::ImplementationType_NONE
    );
 
    ElementOutput expand_operation_dataflow(
        builder::StructuredSDFGBuilder& builder,
        structured_control_flow::Block& block,
        std::vector<ElementInput>& needed_inputs,
        types ::PrimitiveType expected_type
    ) override;
 
    bool supports_integer_types() const override {
        return true;
    }
 
    int tensor_input_count() const override {
        return 1;
    }
 
    void validate_non_tensor_inputs(const data_flow::DataFlowGraph& grap) const override;
 
    std::unique_ptr<data_flow::DataFlowNode>
    clone(size_t element_id, const graph::Vertex vertex, data_flow::DataFlowGraph& parent) const override;
};
 
typedef SimpleElementWiseDataflowTensorNodeSerializer<FillNode> FillNodeSerializer;
 
 
} // namespace tensor
} // namespace math
} // namespace sdfg