Skip to content
Snippets Groups Projects

Add first version of forwardDType.

Merged Cyril Moineau requested to merge ForwardDtype into dev
Files
13
@@ -295,6 +295,8 @@ public:
*/
bool forwardDims(const std::vector<std::vector<DimSize_t>>& dims = {}, bool allowDataDependency = false);
bool forwardDType(const std::vector<DataType>& inputTypes = {});
/** @brief Set the same backend for each Operator of the GraphView object's Nodes. */
void setBackend(const std::string& backend, const DeviceIdx_t device = 0) const;
/** @brief Set the same data type for each Operator of the GraphView object's Nodes. */
@@ -613,6 +615,14 @@ private:
*/
void updateInputsOutputsDelete(NodePtr deletedNode);
/**
* @brief Check if graph connections are valids
*
* @return true
* @return false
*/
bool connectionValid();
///////////////////////////////////////////////////////
// TOPOLOGY
///////////////////////////////////////////////////////
Loading