Searched defs:flat_shapes (Results 1 – 1 of 1) sorted by relevance
842 def _get_defun_inputs_from_args(args, names, flat_shapes=None): argument848 def _get_defun_inputs(args, names, structure, flat_shapes=None): argument945 def _get_defun_inputs_from_kwargs(kwargs, flat_shapes): argument