You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by bg...@apache.org on 2022/01/04 09:03:21 UTC
[incubator-mxnet] 05/06: Add comment
This is an automated email from the ASF dual-hosted git repository.
bgawrych pushed a commit to branch take_opt
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
commit 8a46b85f0d2a07a4112f1313d06bcf7b55e45f13
Author: Bartlomiej Gawrych <ba...@intel.com>
AuthorDate: Mon Nov 29 15:52:02 2021 +0100
Add comment
---
src/operator/tensor/indexing_op.h | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/operator/tensor/indexing_op.h b/src/operator/tensor/indexing_op.h
index 81a04aa..af4b559 100644
--- a/src/operator/tensor/indexing_op.h
+++ b/src/operator/tensor/indexing_op.h
@@ -217,6 +217,7 @@ inline bool EmbeddingOpBackwardStorageType(const nnvm::NodeAttrs& attrs,
/*! \brief name the struct TakeNonzeroAxis for general take when
* axis is not zero, use TakeZeroAxisGPU or TakeZeroAxisCPU for axis zero
+ * or TakeNonZeroAxisCPU for CPU optimized version
*/
template <bool clip = true>
struct TakeNonzeroAxis {