Searched refs:new_function_name (Results 1 – 4 of 4) 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()
886 const auto new_function_name = in PostprocessLiftedArgsForCall() local888 rewritten_fdef.mutable_signature()->set_name(new_function_name); in PostprocessLiftedArgsForCall()897 *node_def.mutable_op() = new_function_name; in PostprocessLiftedArgsForCall()
303 new_function_name, new_args = get_func_and_args_from_str(text)304 if "__internal__" in new_function_name:307 if new_function_name == "tf.compat.v1.%s" % name:313 (new_function_name, text_input, text))315 if new_function_name.startswith("tf.compat.v2"):316 self.assertIn(new_function_name.replace("tf.compat.v2.", "tf."),320 args_v2 = get_args(self.v2_symbols[new_function_name])329 if new_function_name in set(["tf.nn.ctc_loss",332 args_v1 = get_args(self.v1_symbols[new_function_name])
240 std::string new_function_name = GetFunctionImplName( in GetAlternativeViewForSpec() local242 cloned_func.setName(new_function_name); in GetAlternativeViewForSpec()