Searched refs:override_flat_arg_shapes (Results 1 – 2 of 2) sorted by relevance
826 override_flat_arg_shapes=None): argument897 if signature is not None and override_flat_arg_shapes is not None:900 % (signature, override_flat_arg_shapes))907 if override_flat_arg_shapes is not None:909 arg_shapes = override_flat_arg_shapes[:len(flat_args)]910 kwarg_shapes = override_flat_arg_shapes[len(flat_args):]
3238 def _create_graph_function(self, args, kwargs, override_flat_arg_shapes=None): argument3265 override_flat_arg_shapes=override_flat_arg_shapes,3340 args, kwargs, override_flat_arg_shapes=relaxed_arg_shapes)