Skip to content
Snippets Groups Projects

Fix Cast_Op not working

Merged Houssem ROUIS requested to merge fix_cast into dev
1 unresolved thread
Files
2
@@ -123,6 +123,13 @@ public:
*/
inline DataType& targetType() const { return mAttributes->template getAttr<CastAttr::TargetType>(); }
/**
* @brief Sets the data type of the operator's tensors.
* @details This method needs to be overwritten because Cast_Op's output type needs to be set from targetType
* @param dataType Data type to set.
*/
virtual void setDataType(const DataType& dataType) const override;
/**
* @brief Get the input tensor names for the Cast operator.
* @return A vector containing the input tensor names.
Loading