add log message and TODO

apache · Nov 6, 2019 · 14a3791 · 14a3791
1 parent ec1a437
commit 14a3791
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/operator/nn/dropout-inl.h b/src/operator/nn/dropout-inl.h
@@ -402,6 +402,8 @@ class DropoutOp {
                                         this->pkeep_);
           return;
         } else {
+          // TODO (lnyuan) : support axes param
+          LOG(FATAL) << "param axes is not yet supported in this PR";
           RandGenerator<xpu, DType> *pgen = ctx.requested[0].get_parallel_random<xpu, DType>();
           CHECK_NOTNULL(pgen);
           // initialize the mask