Skip to content

Commit

Permalink
Cleanup (#14)
Browse files Browse the repository at this point in the history
  • Loading branch information
MarisaKirisame authored and jroesch committed Aug 16, 2018
1 parent 579a020 commit 53f5ac5
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 14 deletions.
2 changes: 1 addition & 1 deletion relay/include/relay/environment.h
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*!
* Copyright (c) 2017 by Contributors
* Copyright (c) 2018 by Contributors
* \file environment.h
* \brief The global environment containing
*/
Expand Down
21 changes: 9 additions & 12 deletions relay/include/relay/node.h
Original file line number Diff line number Diff line change
Expand Up @@ -421,7 +421,7 @@ TVM_DEFINE_NODE_REF(LocalId, LocalIdNode);

class GlobalId;

/*! \brief A LocalId from the node's current type to target type. */
/*! \brief A GlobalId from the node's current type to target type. */
class GlobalIdNode : public ExprNode {
public:
std::string name;
Expand All @@ -440,7 +440,7 @@ TVM_DEFINE_NODE_REF(GlobalId, GlobalIdNode);

class IntrinsicId;

/*! \brief A LocalId from the node's current type to target type. */
/*! \brief A IntrinsicId from the node's current type to target type. */
class IntrinsicIdNode : public ExprNode {
public:
std::string name;
Expand All @@ -459,7 +459,7 @@ TVM_DEFINE_NODE_REF(IntrinsicId, IntrinsicIdNode)

class Param;

/*! \brief A cast from the node's current type to target type. */
/*! \brief A parameter. */
class ParamNode : public ExprNode {
public:
LocalId id;
Expand All @@ -482,7 +482,7 @@ TVM_DEFINE_NODE_REF(Param, ParamNode)

class Function;

/*! \brief A cast from the node's current type to target type. */
/*! \brief A function. */
class FunctionNode : public ExprNode {
public:
tvm::Array<Param> params;
Expand All @@ -505,7 +505,7 @@ TVM_DEFINE_NODE_REF(Function, FunctionNode)

class Call;

/*! \brief A cast from the node's current type to target type. */
/*! \brief A call. */
class CallNode : public ExprNode {
public:
Expr fn;
Expand All @@ -528,7 +528,7 @@ TVM_DEFINE_NODE_REF(Call, CallNode)

class Debug;

/*! \brief A cast from the node's current type to target type. */
/*! \brief Debug. */
class DebugNode : public ExprNode {
public:
Expr node;
Expand All @@ -551,7 +551,7 @@ enum UOp : int {
NEG = 0,
};

/*! \brief A cast from the node's current type to target type. */
/*! \brief Unary Operator. */
class UnaryOpNode : public ExprNode {
public:
UOp op;
Expand Down Expand Up @@ -587,7 +587,7 @@ enum BinOp : int {
GE = 9
};

/*! \brief A cast from the node's current type to target type. */
/*! \brief A Binary Operator. */
class BinaryOpNode : public ExprNode {
public:
BinOp op;
Expand All @@ -614,7 +614,7 @@ class Assignment;

// TODO(jroesch) : make me contain the proper fields.

/*! \brief A cast from the node's current type to target type. */
/*! \brief Assignment. */
class AssignmentNode : public ExprNode {
public:
tvm::Map<LocalId, Expr> assignments;
Expand Down Expand Up @@ -718,7 +718,6 @@ class ProjectionNode : public ExprNode {

TVM_DEFINE_NODE_REF(Projection, ProjectionNode);

/*! \brief Base type of the Relay item hiearchy. */
struct ItemNode : Node {
public:
GlobalId id;
Expand All @@ -741,7 +740,6 @@ struct Item : public NodeRef {

class Primitive;

/*! \brief A cast from the node's current type to target type. */
class PrimitiveNode : public ItemNode {
public:
Type type;
Expand All @@ -763,7 +761,6 @@ TVM_DEFINE_NODE_REF(Primitive, PrimitiveNode);

class Defn;

/*! \brief A cast from the node's current type to target type. */
class DefnNode : public ItemNode {
public:
Type type;
Expand Down
2 changes: 1 addition & 1 deletion relay/src/relay/environment.cc
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
/*!
* Copyright (c) 2018 by Contributors
* \file node.cc
* \file environment.cc
* \brief Relay node data structure.
*/
#include <nnvm/relay/environment.h>
Expand Down

0 comments on commit 53f5ac5

Please sign in to comment.