src/definitions.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

#pragma once

#include <vector>
#include <string>
#include <functional>

namespace marian {
  typedef float Float;  
  typedef std::vector<int> Shape;
  const int whatevs{-1};
}

#include "keywords.h"
#include "tensor.h"

namespace marian {
  class Tensor;

  namespace keywords {
    KEY(axis, int)
    KEY(name, std::string)
    KEY(shape, Shape)
    KEY(value, float)
    KEY(lazy_shape, std::function<Shape()>)
    KEY(lazy_value, std::function<float()>)
    KEY(init, std::function<void(Tensor)>)
  }

}