sigmoidop.h

Go to the documentation of this file.

#pragma once
#include <string>
#include "compute/operation.h"
#include "compute/add/addop.h"
#include "compute/dot/dotop.h"
#include "compute/variable.h"
#include "compute/negative/negativeop.h"
#include "sigmoid_internal.h"
#include "utilities_internal.h"

namespace magmadnn {
namespace op {

template <typename T>
class SigmoidOp : public Operation<T> {
public:
    SigmoidOp(Operation<T> *x, bool copy=true, bool fast=true);


    std::string to_string() { return "SIGMOID( " + x->to_string() + " )"; }

protected:
    Tensor<T> *_eval(bool recompute=true);
    Tensor<T> *_grad(Operation<T> *consumer, Operation<T> *var, Tensor<T> *grad);

    Operation<T> *x;
    Tensor<T> *x_tensor;
    
    bool copy;
    bool fast;
};

template <typename T>
SigmoidOp<T>* sigmoid(Operation<T> *x, bool copy=true, bool fast=false);

}   // namespace op
}   // namespace magmadnn