#include "train.h"
#include "graph.h"
#include "operation.h"
#include "../../lib/conversions.h"
#include "../tf_ops.h"

namespace addons {

using namespace tf::ops;
using namespace v8;

NAN_MODULE_INIT(Train::Init) {
  Nan::Persistent<Object> inner;
  Local<Object> obj = Nan::New<Object>();
  inner.Reset(obj);

  Nan::SetMethod(obj, "GradientDescentOptimizer", GradientDescentOptimizer);

  target->Set(Nan::New("Train").ToLocalChecked(), obj);
};

NAN_METHOD(Train::GradientDescentOptimizer) {
  TF_Graph* graph = ObjectWrap::Unwrap<addons::Graph>(info[0]->ToObject())->ref();
  float arg0 = info[1]->NumberValue();
  TF_Operation* arg1 = ObjectWrap::Unwrap<Operation>(info[2]->ToObject())->ref(); 
  TF_Operation* result = tf::ops::GradientDescentOptimizer(graph, arg0, arg1);

  info.GetReturnValue().Set((new Operation(result))->ToValue());
}

} // namespace addons