Searched refs:new_function_name (Results 1 – 3 of 3) sorted by relevance
49 const string& xla_cluster_name, const string& new_function_name) in RewriteOutsideCompilationSubgraphFn() argument53 new_function_name_(new_function_name) {} in RewriteOutsideCompilationSubgraphFn()
880 const auto new_function_name = in PostprocessLiftedArgsForCall() local882 rewritten_fdef.mutable_signature()->set_name(new_function_name); in PostprocessLiftedArgsForCall()891 *node_def.mutable_op() = new_function_name; in PostprocessLiftedArgsForCall()
307 new_function_name, new_args = get_func_and_args_from_str(text)308 if new_function_name == "tf.compat.v1.%s" % name:314 (new_function_name, text_input, text))316 if new_function_name.startswith("tf.compat.v2"):317 self.assertIn(new_function_name.replace("tf.compat.v2.", "tf."),321 args_v2 = get_args(self.v2_symbols[new_function_name])330 if new_function_name in set(["tf.nn.ctc_loss",333 args_v1 = get_args(self.v1_symbols[new_function_name])