344         args_chunk_spec: Optional[Tuple[TensorChunkSpec, ...]] = None,
345         kwargs_chunk_spec: Optional[Dict[str, TensorChunkSpec]] = None,  argument
566         args_chunk_spec: Optional[Tuple[TensorChunkSpec, ...]] = None,
567         kwargs_chunk_spec: Optional[Dict[str, TensorChunkSpec]] = None,  argument
1047         args_chunk_spec: Optional[Tuple[TensorChunkSpec, ...]] = None,
1048         kwargs_chunk_spec: Optional[Dict[str, TensorChunkSpec]] = None,  argument
1820         args_chunk_spec: Optional[Tuple[TensorChunkSpec, ...]] = None,
1821         kwargs_chunk_spec: Optional[Dict[str, TensorChunkSpec]] = None,  argument
1928         args_chunk_spec: Optional[Tuple[TensorChunkSpec, ...]] = None,
1929         kwargs_chunk_spec: Optional[Dict[str, TensorChunkSpec]] = None,  argument
2128         args_chunk_spec: Optional[Tuple[TensorChunkSpec, ...]] = None,
2129         kwargs_chunk_spec: Optional[Dict[str, TensorChunkSpec]] = None,  argument