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 <[email protected]> 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 {
