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

[TF2XLA] [NFC] Log which ops require which constant inputs

Browse Source 
PiperOrigin-RevId: 340941883
Change-Id: If6b709124f4b9d055a725560d766dccb9c431ea5
This commit is contained in:
George Karpenkov 2020-11-05 15:40:50 -08:00 committed by TensorFlower Gardener
parent d06e065eb0
commit 0c36726bf2
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/compiler/tf2xla/xla_op_registry.cc
View File

@ -412,6 +412,11 @@ XlaOpRegistry::CompileTimeConstantInputArgNames(const string& op) {
}
}
VLOG(3) << "For operation "
<< (op_def != nullptr ? op_def->name() : op_kernel->name())
<< " required constants are: "
<< absl::StrJoin(*compile_time_constant_inputs, ", ");
for (const string& input : *compile_time_constant_inputs) {
if (op_def) {
NameRangeMap input_name_ranges;