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

Add debug information for _Arg nodes.

Browse Source 
PiperOrigin-RevId: 283813323
Change-Id: I6696b29d6f4fb56af72fde91ee59d5ba6924e0a4
This commit is contained in:
Tong Shen 2019-12-04 12:44:40 -08:00 committed by TensorFlower Gardener
parent 8d3ece77c7
commit ecb57cfd7a
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/compiler/tf2xla/xla_compiler.cc
View File

@ -1059,7 +1059,12 @@ Status XlaCompiler::BuildArguments(
const XlaCompiler::Argument& arg = args[input_to_args->at(i)];
VLOG(2) << " XLA arg " << i
<< " shape: " << xla::ShapeUtil::HumanString(arg_shapes[i])
<< " name: " << arg.name << " TF arg " << input_to_args->at(i);
<< " name: " << arg.name << " TF arg " << input_to_args->at(i)
<< " node name: " << arg.node_name
<< (arg_shardings.find(i) == arg_shardings.end()
? ""
: absl::StrCat(" sharding: ",
arg_shardings.at(i).DebugString()));
XlaExpression& arg_expression = (*arg_expressions)[input_to_args->at(i)];
switch (arg.kind) {
case XlaCompiler::Argument::kResource: {

3
tensorflow/compiler/tf2xla/xla_compiler.h
View File

@ -147,6 +147,9 @@ class XlaCompiler {
// The name of this argument, used for debugging.
string name;
// The name of TensorFlow _Arg node, used for debugging.
string node_name;
// For a kResource, what kind of resource is it?
XlaResource::Kind resource_kind = XlaResource::kInvalid;