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

Fix ASan error while accessing opcode->custom_name.

Browse Source 
PiperOrigin-RevId: 251879003
This commit is contained in:
Sachin Joglekar 2019-06-06 10:35:30 -07:00 committed by TensorFlower Gardener
parent a372bb0e9d
commit dada3c3688
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/lite/model.cc
View File

@ -228,17 +228,23 @@ TfLiteStatus InterpreterBuilder::BuildLocalIndexToRegistrationMapping() {
}
// If it's an unresolved custom op, allow it for now. It might be resolved
// by a delegate later.
if (!opcode->custom_code()) {
error_reporter_->Report(
"Operator with CUSTOM builtin_code has no custom_code.\n");
return status;
}
const auto* op_name = opcode->custom_code()->c_str();
TfLiteRegistration unresolved_op{nullptr,
nullptr,
nullptr,
/*invoke*/ &UnresolvedOpInvoke,
nullptr,
BuiltinOperator_CUSTOM,
opcode->custom_code()->c_str(),
op_name,
1};
unresolved_custom_ops_.push_back(unresolved_op);
registration = &unresolved_custom_ops_.back();
has_flex_op_ |= IsFlexOp(registration->custom_name);
has_flex_op_ |= IsFlexOp(op_name);
status = kTfLiteOk;
}
flatbuffer_op_index_to_registration_.push_back(registration);