gradients.h

Go to the documentation of this file.

#pragma once

#include <vector>
#include "compute/operation.h"
#include "compute/variable.h"
#include "compute/gradtable.h"
#include "compute/variable.h"
#include "compute/add/addop.h"
#include "compute/sum/sumop.h"
#include "utilities_internal.h"

namespace magmadnn {
namespace op {

template <typename T>
magmadnn_error_t get_grad_table(const std::vector<Operation<T> *>& vars, Operation<T> *graph, GradTable<T> &table);

} // namespace op

// build_grad should only be used internally
namespace internal {

template <typename T>
magmadnn_error_t build_grad(op::Operation<T>* var, op::Operation<T> *graph, op::GradTable<T> &table, Tensor<T> **grad);

}   // namespace internal
}   // namespace magmadnn