Searched refs:override_flat_arg_shapes (Results 1 – 2 of 2) sorted by relevance
528 override_flat_arg_shapes=None): argument598 if signature is not None and override_flat_arg_shapes is not None:601 % (signature, override_flat_arg_shapes))608 if override_flat_arg_shapes is not None:610 arg_shapes = override_flat_arg_shapes[:len(flat_args)]611 kwarg_shapes = override_flat_arg_shapes[len(flat_args):]
1465 def _create_graph_function(self, args, kwargs, override_flat_arg_shapes=None): argument1490 override_flat_arg_shapes=override_flat_arg_shapes,1589 args, kwargs, override_flat_arg_shapes=relaxed_arg_shapes)