experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #36856 from wwwind:crash_conv_dilation

Browse Source 
PiperOrigin-RevId: 296249506
Change-Id: Ic6401824fab91257354e8ae3dfbbdba505304f3c
This commit is contained in:
TensorFlower Gardener 2020-02-20 10:59:58 -08:00
commit a165c7fa08
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/lite/toco/graph_transformations/identify_dilated_conv.cc
View File

@ -86,7 +86,7 @@ bool ResolveDilatedConv(Model* model, Operator* conv_base_op, Operator* stb_op,
? GetOpWithInput(*model, post_conv_op->outputs[0]) ? GetOpWithInput(*model, post_conv_op->outputs[0])
: GetOpWithInput(*model, conv_op->outputs[0]); : GetOpWithInput(*model, conv_op->outputs[0]);
bool has_pad_op = false; bool has_pad_op = false;
if (pad_op->type == OperatorType::kPad) { if (pad_op && pad_op->type == OperatorType::kPad) {
has_pad_op = true; has_pad_op = true;
CHECK_EQ(pad_op->inputs.size(), 2); CHECK_EQ(pad_op->inputs.size(), 2);
CHECK_EQ(pad_op->outputs.size(), 1); CHECK_EQ(pad_op->outputs.size(), 1);
@ -128,7 +128,7 @@ bool ResolveDilatedConv(Model* model, Operator* conv_base_op, Operator* stb_op,
if (!has_pad_op) { if (!has_pad_op) {
auto* pre_stb_pad_op = GetOpWithOutput(*model, stb_op->inputs[0]); auto* pre_stb_pad_op = GetOpWithOutput(*model, stb_op->inputs[0]);
// If it is a Pad Op then just rewire the Input of Pad Op with Input of STB // If it is a Pad Op then just rewire the Input of Pad Op with Input of STB
if (pre_stb_pad_op->type == OperatorType::kPad) { if (pre_stb_pad_op && pre_stb_pad_op->type == OperatorType::kPad) {
stb_op->inputs[0] = pre_stb_pad_op->inputs[0]; stb_op->inputs[0] = pre_stb_pad_op->inputs[0];
has_pad_op = true; has_pad_op = true;
pad_op = pre_stb_pad_op; pad_op = pre_stb_pad_op;