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

Don't return an uninitialized value from TFE_OpNameGetAttrType

Browse Source 
PiperOrigin-RevId: 262251730
This commit is contained in:
Akshay Modi 2019-08-07 17:36:34 -07:00 committed by Akshay Naresh Modi
parent ff98617eb0
commit d32cc881cf
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/c/eager/c_api.cc
View File

@ -671,7 +671,7 @@ void TFE_OpAddInputList(TFE_Op* op, TFE_TensorHandle** inputs, int num_inputs,
TF_AttrType TFE_OpGetAttrType(TFE_Op* op, const char* attr_name,
unsigned char* is_list, TF_Status* status) {
TF_AttrType ret;
TF_AttrType ret = TF_ATTR_INT;
status->status = tensorflow::AttrTypeByName(*op->operation.AttrTypes(),
attr_name, &ret, is_list);
return ret;

2
tensorflow/python/ops/while_v2.py
View File

@ -340,7 +340,7 @@ def _WhileGrad(op, *grads): # pylint: disable=invalid-name
parallel_iterations = op.get_attr("parallel_iterations")
try:
num_original_outputs = op.get_attr("_num_original_outputs")
num_original_outputs = while_op.get_attr("_num_original_outputs")
except: # pylint: disable=bare-except
num_original_outputs = len(while_op.outputs)

1
tensorflow/python/pywrap_tfe.i
View File

@ -233,6 +233,7 @@ static PyObject* TFE_ClearScalarCache();
}
%typemap(in, numinputs=0) unsigned char* is_list (unsigned char tmp) {
tmp = 0;
$1 = &tmp;
}