Searched refs:split_args_kwargs_into_chunks (Results 1 – 4 of 4) sorted by relevance
9 split_args_kwargs_into_chunks,29 arg_chunks, kwarg_chunks = split_args_kwargs_into_chunks(args, kwargs, 2)70 args_split, kwargs_split = split_args_kwargs_into_chunks(
246 def split_args_kwargs_into_chunks( function
29 from .microbatch import merge_chunks, split_args_kwargs_into_chunks, TensorChunkSpec492 args_split, kwargs_split = split_args_kwargs_into_chunks(
460 .. autofunction:: split_args_kwargs_into_chunks