Searched refs:_all_to_all_abstract_eval (Results 1 – 1 of 1) sorted by relevance
774 def _all_to_all_abstract_eval(x, axis_name, split_axis, concat_axis, axis_index_groups): function782 all_to_all_p.def_abstract_eval(_all_to_all_abstract_eval)