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

Print node name in GatherNdSlice error msg.

Browse Source 
PiperOrigin-RevId: 346726189
Change-Id: Idb251f7e8f8cd28fd2998bcecdd9b022f61dfd09
This commit is contained in:
James Qin 2020-12-10 00:27:22 -08:00 committed by TensorFlower Gardener
parent 51cd440663
commit c046c38f6f
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/core/kernels/gather_nd_op.h
View File

@ -161,7 +161,8 @@ Status DoGatherNd(OpKernelContext* c, const Tensor& params,
"indices", SliceDebugString(shape, bad_i), " = [", "indices", SliceDebugString(shape, bad_i), " = [",
str_util::Join( str_util::Join(
gtl::ArraySlice<Index>(&indices_mat(bad_i, 0), indices_nd), ", "), gtl::ArraySlice<Index>(&indices_mat(bad_i, 0), indices_nd), ", "),
"] does not index into param shape ", params.shape().DebugString()); "] does not index into param shape ", params.shape().DebugString(),
", node name: ", c->op_kernel().name());
} }
} }
return Status::OK(); return Status::OK();