#ifndef NNCC_IR_NODE_DOT_BUILDER_H
#define NNCC_IR_NODE_DOT_BUILDER_H
-#include "nncc/core/ADT/tensor/Shape.h"
+#include "nnc/core/linalg/Shape.h"
#include "nnc/core/IR/model/operations/common.h"
#include "nnc/core/IR/model/operations/pool_op.h"
{
using namespace nncc::contrib::core::IR::model;
-using nncc::core::ADT::tensor::Shape;
+using namespace nncc::contrib::core::data;
/**
* @brief Can collect information about a NN operator, and then use it to output
#include <iostream>
-#include "nncc/core/ADT/tensor/Shape.h"
+#include "nnc/core/linalg/Shape.h"
#include "nnc/core/IR/model/graph/ir_node.h"
#include "nnc/core/IR/dumpers/ir_dot_node_info.h"
namespace dumper
{
-using nncc::core::ADT::tensor::Shape;
+using namespace nncc::contrib::core::data;
static std::vector<Shape> getInputShapes(OpDescription &op)
{
namespace dumper
{
-std::ostream &operator<<(std::ostream &s, const Shape &shape) noexcept
-{
- s << "[";
- for (uint32_t d = 0; d < shape.rank(); ++d)
- {
- if (d != 0)
- s << ", ";
- s << shape.dim(d);
- }
- s << "]";
- return s;
-}
-
DotIrNodeInfo &DotIrNodeInfo::withType(const std::string &typeName, const std::string &nodeName)
{
this->typeName = typeName;