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

Including attributes in the error message in case of attributes mismatch.

Browse Source 
PiperOrigin-RevId: 221870891
This commit is contained in:
A. Unique TensorFlower 2018-11-16 16:25:04 -08:00 committed by TensorFlower Gardener
parent de8ebdb12f
commit db1b3aafc0
3 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/framework/node_def_util.cc
View File

@ -102,6 +102,10 @@ string SummarizeNodeDef(const NodeDef& node_def) {
return ret; return ret;
} }
string SummarizeAttrs(const NodeDef& node_def) {
return SummarizeAttrsHelper(node_def, node_def.device());
}
string FormatNodeForError(const Node& node) { string FormatNodeForError(const Node& node) {
return FormatNodeDefForError(node.def()); return FormatNodeDefForError(node.def());
} }

1
tensorflow/core/framework/node_def_util.h
View File

@ -48,6 +48,7 @@ extern const char* const kColocationGroupPrefix;
// than a text-format proto. // than a text-format proto.
string SummarizeNode(const Node& node); string SummarizeNode(const Node& node);
string SummarizeNodeDef(const NodeDef& node_def); string SummarizeNodeDef(const NodeDef& node_def);
string SummarizeAttrs(const NodeDef& node_def);
// Produces a formatted string pattern from the node which can uniquely identify // Produces a formatted string pattern from the node which can uniquely identify
// this node upstream to produce an informative error message. The pattern // this node upstream to produce an informative error message. The pattern

6
tensorflow/core/framework/op_kernel.cc
View File

@ -1127,7 +1127,8 @@ Status FindKernelDef(const DeviceType& device_type, const NodeDef& node_def,
FormatNodeDefForError(node_def)); FormatNodeDefForError(node_def));
if (was_attr_mismatch) { if (was_attr_mismatch) {
errors::AppendToMessage( errors::AppendToMessage(
&s, " (OpKernel was found, but attributes didn't match)"); &s, " (OpKernel was found, but attributes didn't match) ",
"Requested Attributes: ", SummarizeAttrs(node_def));
} }
errors::AppendToMessage( errors::AppendToMessage(
&s, ". Registered:", KernelsRegisteredForOp(node_def.op())); &s, ". Registered:", KernelsRegisteredForOp(node_def.op()));
@ -1262,7 +1263,8 @@ Status CreateOpKernel(DeviceType device_type, DeviceBase* device,
FormatNodeDefForError(node_def))); FormatNodeDefForError(node_def)));
if (was_attr_mismatch) { if (was_attr_mismatch) {
errors::AppendToMessage( errors::AppendToMessage(
&s, " (OpKernel was found, but attributes didn't match)"); &s, " (OpKernel was found, but attributes didn't match) ",
"Requested Attributes: ", SummarizeAttrs(node_def));
} }
errors::AppendToMessage( errors::AppendToMessage(
&s, ". Registered:", KernelsRegisteredForOp(node_def.op())); &s, ". Registered:", KernelsRegisteredForOp(node_def.op()));