chore: import upstream snapshot with attribution
This commit is contained in:
wehub-resource-sync
@@ -0,0 +1,232 @@
|
||||
/* ******************************************************************************
|
||||
*
|
||||
*
|
||||
* This program and the accompanying materials are made available under the
|
||||
* terms of the Apache License, Version 2.0 which is available at
|
||||
* https://www.apache.org/licenses/LICENSE-2.0.
|
||||
*
|
||||
* See the NOTICE file distributed with this work for additional
|
||||
* information regarding copyright ownership.
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||||
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||||
* License for the specific language governing permissions and limitations
|
||||
* under the License.
|
||||
*
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
******************************************************************************/
|
||||
|
||||
//
|
||||
// @author raver119@gmail.com
|
||||
//
|
||||
|
||||
#ifndef LIBND4J_GNODE_H
|
||||
#define LIBND4J_GNODE_H
|
||||
#include <array/NDArray.h>
|
||||
#include <graph/generated/node_generated.h>
|
||||
#include <ops/declarable/DeclarableOp.h>
|
||||
|
||||
#include <atomic>
|
||||
#include <string>
|
||||
|
||||
#include "Context.h"
|
||||
#include "array/NDArray.h"
|
||||
|
||||
namespace sd {
|
||||
namespace graph {
|
||||
|
||||
class Graph;
|
||||
|
||||
class SD_LIB_EXPORT Node {
|
||||
protected:
|
||||
// TODO: this field must be removed
|
||||
DataType _dataType;
|
||||
|
||||
::graph::OpType _opType;
|
||||
ContextPrototype *_protoContext = nullptr;
|
||||
LongType _opNum;
|
||||
int _id;
|
||||
std::vector<std::pair<int, int>> _input;
|
||||
std::vector<std::pair<int, int>> _output;
|
||||
std::vector<LongType> _dimensions;
|
||||
|
||||
std::vector<int> _referencedBy;
|
||||
|
||||
LongType *_dim = nullptr;
|
||||
std::string _name;
|
||||
|
||||
// this variable points to onion layer within graph
|
||||
int _layer = -1;
|
||||
|
||||
// many ops require extra parameters to run
|
||||
double *_extraParams = nullptr;
|
||||
|
||||
|
||||
|
||||
bool _hasExternalOutputs;
|
||||
bool _hasExternalInputs;
|
||||
bool _hasInternalOutputs;
|
||||
bool _hasInternalInputs;
|
||||
|
||||
// this field is used to check, if op should be used in-place (so it can/will modify its inputs)
|
||||
bool _isInplace = false;
|
||||
|
||||
// this field is used to delete attached customOp
|
||||
bool _isDeductable = false;
|
||||
|
||||
::graph::OpClass _opClass;
|
||||
|
||||
// these fields are used to store embedded CustomOps and Graph in case of Graph-in-Graph scenario
|
||||
Graph *_graph = nullptr;
|
||||
ops::DeclarableOp *_customOp = nullptr;
|
||||
|
||||
// each node can be active or inactive, if used with divergents, like IF statements
|
||||
bool _active = true;
|
||||
|
||||
// these fields contain information about OpScope these ops are related to
|
||||
int _scope_id = 0;
|
||||
std::string _scope_name;
|
||||
|
||||
// TODO: these 3 fields should be removed
|
||||
int _rewindNode = -1;
|
||||
std::pair<int, int> _rewindLayer = {-1, -1};
|
||||
LongType _frameId = -1;
|
||||
|
||||
public:
|
||||
explicit Node(ops::DeclarableOp *customOp, int id = 0, std::initializer_list<int> input = {},
|
||||
std::initializer_list<int> output = {}, std::initializer_list<int> dimensions = {}, float scalar = 0.0f,
|
||||
std::initializer_list<double> tArgs = {}, std::initializer_list<int> iArgs = {});
|
||||
explicit Node(::graph::OpType opType = ::graph::OpType_TRANSFORM_SAME, int opNum = 0, int id = 0, std::initializer_list<int> input = {},
|
||||
std::initializer_list<int> output = {}, std::initializer_list<int> dimensions = {}, float scalar = 0.0f,
|
||||
std::initializer_list<double> tArgs = {}, std::initializer_list<int> iArgs = {});
|
||||
explicit Node(const ::graph::FlatNode *node);
|
||||
~Node();
|
||||
|
||||
bool equals(Node *other);
|
||||
|
||||
DataType dataType();
|
||||
ContextPrototype *protoContext();
|
||||
::graph::OpType opType();
|
||||
LongType opNum();
|
||||
int id();
|
||||
std::vector<std::pair<int, int>> *input();
|
||||
std::vector<std::pair<int, int>> *output();
|
||||
|
||||
LongType getFrameId();
|
||||
void setFrameId(LongType frameId);
|
||||
|
||||
int getRewindNode();
|
||||
void setRewindNode(int nodeId);
|
||||
|
||||
std::pair<int, int> &getRewindLayer();
|
||||
void setRewindLayer(int layerId, int stepId = 0);
|
||||
|
||||
void setId(int id);
|
||||
|
||||
double *extraParams();
|
||||
|
||||
bool isMultiInput();
|
||||
bool isMultiOutput();
|
||||
|
||||
int getLayer();
|
||||
void setLayer(int layer);
|
||||
|
||||
bool isDivergencePoint();
|
||||
void setActive(bool reallyActive);
|
||||
bool isActive();
|
||||
|
||||
bool hasExternalOutputs();
|
||||
bool hasExternalInputs();
|
||||
bool hasInternalOutputs();
|
||||
bool hasInternalInputs();
|
||||
|
||||
std::vector<LongType> *getDimensions();
|
||||
LongType *getDimensionsPtr();
|
||||
|
||||
void pickOutputOnce(int outputId);
|
||||
void pickOutput(int outputId);
|
||||
void pickOutput(int nodeId, int outputId);
|
||||
void pickExternalOutput(int outputId);
|
||||
void pickInput(int inputId);
|
||||
void pickInput(int nodeId, int outputId);
|
||||
void pickInput(std::pair<int, int> &id);
|
||||
|
||||
bool isDeductable();
|
||||
void setDeductable(bool reallyDeductable);
|
||||
|
||||
void setName(std::string *name);
|
||||
void setName(const std::string &name);
|
||||
std::string *getName();
|
||||
std::string *name();
|
||||
|
||||
int totalReferences();
|
||||
void addReference(int nodeId);
|
||||
|
||||
void setContextPrototype(ContextPrototype *block);
|
||||
ContextPrototype *getContextPrototype();
|
||||
bool hasBlockAttached();
|
||||
|
||||
void setCustomOp(ops::DeclarableOp *customOp = nullptr);
|
||||
ops::DeclarableOp *getCustomOp();
|
||||
bool hasCustomOp();
|
||||
|
||||
void setGraph(Graph *graph = nullptr);
|
||||
Graph *getGraph();
|
||||
bool hasGraphEmbedded();
|
||||
|
||||
bool isInplace();
|
||||
void markInplace(bool reallyInplace);
|
||||
|
||||
::graph::OpClass getOpClass();
|
||||
|
||||
// these methods are used for internal profiling
|
||||
void setOuterTime(LongType time);
|
||||
void setInnerTime(LongType time);
|
||||
|
||||
// methods related to scopes
|
||||
bool isScoped();
|
||||
void setScopeInfo(int id, const char *name = nullptr);
|
||||
int scopeId();
|
||||
std::string *scopeName();
|
||||
|
||||
void setOpType(::graph::OpType opType);
|
||||
|
||||
// clone Node
|
||||
Node *clone();
|
||||
|
||||
template <typename T>
|
||||
Node *asT();
|
||||
|
||||
SD_INLINE void pullValues(Node *other) {
|
||||
//if (this->_protoContext != nullptr) delete _protoContext;
|
||||
|
||||
this->_dataType = other->dataType();
|
||||
this->_protoContext = other->protoContext()->clone();
|
||||
this->_hasExternalInputs = other->hasExternalInputs();
|
||||
this->_hasExternalOutputs = other->hasExternalOutputs();
|
||||
this->_hasInternalInputs = other->hasInternalInputs();
|
||||
this->_hasInternalOutputs = other->hasInternalOutputs();
|
||||
|
||||
this->markInplace(other->isInplace());
|
||||
this->setActive(other->isActive());
|
||||
this->setScopeInfo(other->scopeId(), other->scopeName()->c_str());
|
||||
this->setLayer(other->getLayer());
|
||||
this->setDeductable(other->isDeductable());
|
||||
|
||||
if (this->_customOp != nullptr && _isDeductable) delete this->_customOp;
|
||||
|
||||
for (auto v : *other->input()) this->_input.emplace_back(v);
|
||||
|
||||
for (auto v : *other->output()) this->_output.emplace_back(v);
|
||||
|
||||
for (auto v : *other->getDimensions()) this->_dimensions.emplace_back(v);
|
||||
}
|
||||
|
||||
static ops::DeclarableOp *buildOpByType(::graph::OpType opType, int numInputs, int numIArgs, int numTArgs, int opNum,
|
||||
NDArray *scalar);
|
||||
static void deleteOpByType(::graph::OpType opType, void *op);
|
||||
};
|
||||
} // namespace graph
|
||||
} // namespace sd
|
||||
|
||||
#endif // LIBND4J_GNODE_H
|
||||
Reference in New Issue
Block a user