Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dtf2xla_util.cc794 const string& rewritten_function_name) { in RewriteAssociatedFunction() argument
799 NodeDefBuilder builder(node->name(), rewritten_function_name, fld, in RewriteAssociatedFunction()
832 gradient_def.set_gradient_func(rewritten_function_name); in RewriteAssociatedFunction()
836 } else if (original_grad_func != rewritten_function_name) { in RewriteAssociatedFunction()
852 func.set_name(rewritten_function_name); in RewriteAssociatedFunction()
H A Dtf2xla_util.h145 const string& rewritten_function_name);