Home
last modified time | relevance | path

Searched refs:rewritten_function_name (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc624 const string& rewritten_function_name) { in RewriteAssociatedFunction() argument
629 NodeDefBuilder builder(node->name(), rewritten_function_name, fld, in RewriteAssociatedFunction()
664 gradient_def.set_gradient_func(rewritten_function_name); in RewriteAssociatedFunction()
668 } else if (original_grad_func != rewritten_function_name) { in RewriteAssociatedFunction()
679 func.set_name(rewritten_function_name); in RewriteAssociatedFunction()
Dtf2xla_util.h143 const string& rewritten_function_name);