Skip to content

GitLab

Commit 805b6f09 authored by Khalique's avatar Khalique
Browse files

formatting

parent ec794967
Show whitespace changes
Inline Side-by-side
Showing with 20 additions and 8 deletions
src/tf/tf.cpp
View file @ 805b6f09
...@@ -317,13 +317,25 @@ struct tf_parser ...@@ -317,13 +317,25 @@ struct tf_parser
size_t weight_w = weight_dims[3]; size_t weight_w = weight_dims[3];
if(pad_mode.find("SAME") != std::string::npos) if(pad_mode.find("SAME") != std::string::npos)
{ {
op.padding[0] = static_cast<size_t>(std::ceil(static_cast<double>(-op.stride[0] + op.dilation[0]*(weight_h-1)+1))/2); op.padding[0] =
op.padding[1] = static_cast<size_t>(std::ceil(static_cast<double>(-op.stride[1] + op.dilation[1]*(weight_w-1)+1))/2); static_cast<size_t>(std::ceil(static_cast<double>(
-op.stride[0] + op.dilation[0] * (weight_h - 1) + 1)) /
2);
op.padding[1] =
static_cast<size_t>(std::ceil(static_cast<double>(
-op.stride[1] + op.dilation[1] * (weight_w - 1) + 1)) /
2);
} }
else if(pad_mode.find("VALID") != std::string::npos) else if(pad_mode.find("VALID") != std::string::npos)
{ {
op.padding[0] = static_cast<size_t>(std::ceil(static_cast<double>(-weight_h - op.stride[0] + op.dilation[0]*(weight_h-1)+1))/2); op.padding[0] = static_cast<size_t>(
op.padding[1] = static_cast<size_t>(std::ceil(static_cast<double>(-weight_w - op.stride[1] + op.dilation[1]*(weight_w-1)+1))/2); std::ceil(static_cast<double>(-weight_h - op.stride[0] +
op.dilation[0] * (weight_h - 1) + 1)) /
2);
op.padding[1] = static_cast<size_t>(
std::ceil(static_cast<double>(-weight_w - op.stride[1] +
op.dilation[1] * (weight_w - 1) + 1)) /
2);
} }
else if(pad_mode.find("EXPLICIT") != std::string::npos) else if(pad_mode.find("EXPLICIT") != std::string::npos)
{ {
......
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