Method Algorithm

src/model/layer/cudnn_softmax.h:46–46

                                                   const Tensor& grad) override;

  const cudnnSoftmaxAlgorithm_t Algorithm() const { return algorithm_; }

 private:
  /// Init cudnn related data structures.