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