elvin-n commented on code in PR #11357:
URL: https://github.com/apache/tvm/pull/11357#discussion_r901946068
##########
include/tvm/te/operation.h:
##########
@@ -76,6 +78,11 @@ class TVM_DLL OperationNode : public Object {
* \return type of i-th output.
*/
virtual DataType output_dtype(size_t i) const = 0;
+ /**
+ * @brief Returns memory scope of operation
+ * TODO(amalyshe): add support for individual output's tensors, not the only
one
+ */
+ String memory_scope() const { return memory_scope_; }
Review Comment:
Done
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
