Commit eb416ae1 authored by Khalique's avatar Khalique
Browse files

fix enums for other operators

parent 019bc600
...@@ -140,12 +140,7 @@ struct im2col ...@@ -140,12 +140,7 @@ struct im2col
std::array<std::size_t, 2> padding = {{0, 0}}; std::array<std::size_t, 2> padding = {{0, 0}};
std::array<std::size_t, 2> stride = {{1, 1}}; std::array<std::size_t, 2> stride = {{1, 1}};
std::array<std::size_t, 2> dilation = {{1, 1}}; std::array<std::size_t, 2> dilation = {{1, 1}};
enum padding_mode_t
{
default_, // NOLINT
same,
valid
};
padding_mode_t padding_mode = default_; padding_mode_t padding_mode = default_;
template <class Self, class F> template <class Self, class F>
...@@ -654,13 +649,13 @@ struct pad ...@@ -654,13 +649,13 @@ struct pad
{ {
std::vector<int64_t> pads; std::vector<int64_t> pads;
float value = 0.0f; float value = 0.0f;
enum padding_mode_t enum pad_op_mode_t
{ {
constant_pad, constant_pad,
reflect_pad, reflect_pad,
edge_pad edge_pad
}; };
padding_mode_t mode = constant_pad; pad_op_mode_t mode = constant_pad;
template <class Self, class F> template <class Self, class F>
static auto reflect(Self& self, F f) static auto reflect(Self& self, F f)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment