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()
879 const auto new_function_name = in PostprocessLiftedArgsForCall() local881 rewritten_fdef.mutable_signature()->set_name(new_function_name); in PostprocessLiftedArgsForCall()890 *node_def.mutable_op() = new_function_name; in PostprocessLiftedArgsForCall()
223 std::string new_function_name = GetFunctionImplName( in GetAlternativeViewForSpec() local225 cloned_func.setName(new_function_name); in GetAlternativeViewForSpec()
309 new_function_name, new_args = get_func_and_args_from_str(text)310 if "__internal__" in new_function_name:313 if new_function_name == "tf.compat.v1.%s" % name:319 (new_function_name, text_input, text))321 if new_function_name.startswith("tf.compat.v2"):322 self.assertIn(new_function_name.replace("tf.compat.v2.", "tf."),326 args_v2 = get_args(self.v2_symbols[new_function_name])335 if new_function_name in set(["tf.nn.ctc_loss",338 args_v1 = get_args(self.v1_symbols[new_function_name])