MCPcopy Create free account

hub / github.com/ml-explore/mlx-data / functions

Functions662 in github.com/ml-explore/mlx-data

Methodbuild_extension
(self, ext)
setup.py:44
Methodcan_merge
mlx/data/core/BPETokenizer.h:23
Methodcancel_prefetch
mlx/data/core/FileFetcher.cpp:143
Methodcheck_credentials
mlx/data/core/AWSFileFetcher.cpp:138
Methodcheck_samples_
mlx/data/buffer/FromVector.cpp:36
Methodcollate_fn
(samples)
benchmarks/comparative/librispeech/pytorch.py:36
Methodconcretize
mlx/data/Buffer.cpp:124
Methodcontains
mlx/data/core/TARReader.cpp:179
Methodcsv_reader_from_key
mlx/data/Stream.cpp:53
Methoddecode_
mlx/data/core/video/VideoFFMPEG.cpp:441
Functiondl_iterate_callback
mlx/data/core/ThreadController.cpp:24
Functiondocument_iterator
Extract documents from wikitext raw file.
benchmarks/comparative/wikitext/tfds.py:12
Functiondocument_iterator
Extract documents from wikitext raw file.
benchmarks/comparative/wikitext/mlx_data.py:13
Methoddot_graph
mlx/data/core/Graph.cpp:32
Methoddynamic_batch
mlx/data/Stream.cpp:75
Methoddynamic_batch_
mlx/data/buffer/DynamicBatch.cpp:68
Functionfile_to_label
(x)
python/mlx/data/datasets/imagenet.py:223
Functionfiles_and_classes
(root: Path)
benchmarks/comparative/librispeech/tfds.py:13
Functionfiles_from_tar
mlx/data/Buffer.cpp:139
Methodfill_queue_
mlx/data/core/FileFetcher.cpp:23
Methodfilter_by_shape
mlx/data/Dataset.cpp:37
Methodfilter_by_shape_if
mlx/data/Dataset.cpp:46
Methodfilter_key_if
mlx/data/Dataset.cpp:65
Methodfind_from_start_node
mlx/data/core/Graph.cpp:261
Methodfind_from_start_node
mlx/data/core/Graph.cpp:304
Methodfinish_background_tasks
mlx/data/stream/Buffered.cpp:99
Methodforward_
mlx/data/core/Tokenizer.cpp:207
Functionframes
Return the frame count of the image. Requires that `verifyVideo()` be called previously.
mlx/data/core/video/Video.h:52
Methodgenerate_random_crop_
mlx/data/op/ImageTransform.cpp:135
Methodget
mlx/data/buffer/FilesFromTAR.cpp:20
Methodget
mlx/data/buffer/Transform.cpp:21
Methodget
mlx/data/buffer/Buffer.cpp:11
Methodget
mlx/data/buffer/Append.cpp:14
Methodget
mlx/data/buffer/Batch.cpp:52
Methodget
mlx/data/buffer/Partition.cpp:27
Methodget_size
mlx/data/core/AWSFileFetcher.cpp:184
Methodget_tar_reader_
mlx/data/op/ReadFromTAR.cpp:32
Functiongpt2_byte_map
Return the byte map used for preprocessing the text for GPT-2 tokenizers. See https://github.com/openai/gpt-2/blob/master/src/encoder.py for deta
python/mlx/data/tokenizer_helpers.py:266
Functionheight
Return the height of the image. Requires that `verifyVideo()` be called previously.
mlx/data/core/video/Video.h:40
Methodimage_center_crop_if
mlx/data/Dataset.cpp:84
Methodimage_channel_reduction
mlx/data/Dataset.cpp:98
Methodimage_channel_reduction_if
mlx/data/Dataset.cpp:107
Methodimage_random_area_crop
mlx/data/Dataset.cpp:121
Methodimage_random_area_crop_if
mlx/data/Dataset.cpp:132
Methodimage_random_crop
mlx/data/Dataset.cpp:148
Methodimage_random_crop_if
mlx/data/Dataset.cpp:157
Methodimage_random_h_flip
mlx/data/Dataset.cpp:171
Methodimage_random_h_flip_if
mlx/data/Dataset.cpp:179
Methodimage_resize
mlx/data/Dataset.cpp:192
Methodimage_resize_if
mlx/data/Dataset.cpp:201
Methodimage_resize_smallest_side_if
mlx/data/Dataset.cpp:224
Methodimage_rotate
mlx/data/Dataset.cpp:238
Methodimage_rotate_if
mlx/data/Dataset.cpp:247
Methodimemstream
mlx/data/core/imemstream.h:18
Functioninfo
mlx/data/core/image/ImageIO.cpp:26
Functioninfo
mlx/data/core/video/Video.cpp:59
Methodinit
mlx/data/core/AWSFileFetcher.cpp:20
Methodinit
finish initializing the `VideoReaderState`. Requires that the `formatContext_` be set.
mlx/data/core/video/VideoFFMPEG.cpp:263
Methodinit_
mlx/data/Array.cpp:69
Methodinit_
mlx/data/stream/LineReader.cpp:27
Methodis_final_node
mlx/data/core/Graph.cpp:157
Methodis_start_node
mlx/data/core/Graph.cpp:154
Methodis_valid_file
(self, filepath: str)
benchmarks/comparative/caltech101/pytorch.py:26
Functioniterate
(args, workers)
benchmarks/comparative/caltech101/pytorch.py:30
Functioniterate
(args, workers)
benchmarks/comparative/caltech101/tfds.py:47
Functioniterate
(args, workers)
benchmarks/comparative/caltech101/mlx_data.py:23
Functioniterate
(args, workers)
benchmarks/comparative/librispeech/pytorch.py:57
Functioniterate
(args, workers)
benchmarks/comparative/librispeech/tfds.py:63
Functioniterate
(args, workers)
benchmarks/comparative/librispeech/mlx_data.py:28
Functioniterate
(args, workers)
benchmarks/comparative/wikitext/pytorch.py:71
Functioniterate
(args)
benchmarks/comparative/wikitext/tfds.py:51
Functioniterate
(args, workers)
benchmarks/comparative/wikitext/mlx_data.py:41
Functionjpeg_error_custom_noop
mlx/data/core/Version.cpp:95
Functionjpeg_mem_fill_input_buffer
mlx/data/core/image/ImageJPEG.cpp:27
Functionjpeg_mem_init_source
mlx/data/core/image/ImageJPEG.cpp:25
Functionjpeg_mem_skip_input_data
mlx/data/core/image/ImageJPEG.cpp:35
Functionjpeg_mem_term_source
mlx/data/core/image/ImageJPEG.cpp:45
Functionlevenshtein_arr
mlx/data/core/Levenshtein.cpp:66
Functionlibs_version
mlx/data/core/Version.cpp:242
Functionload
mlx/data/core/image/ImageIO.cpp:10
Functionload
mlx/data/core/video/Video.cpp:10
Functionload_cifar10
Load a buffer with the CIFAR-10 dataset. Args: root (Path or str, optional): The directory to load/save the data. If none is
python/mlx/data/datasets/cifar.py:105
Functionload_cifar100
Load a buffer with the CIFAR-100 dataset. Args: root (Path or str, optional): The directory to load/save the data. If none is
python/mlx/data/datasets/cifar.py:128
Functionload_fashion_mnist
Load a buffer with the Fashion-MNIST dataset. If the data doesn't exist download it and save it for the next time. Args: root (Path
python/mlx/data/datasets/mnist.py:84
Methodload_file
mlx/data/Dataset.cpp:392
Methodload_file_if
mlx/data/Dataset.cpp:400
Methodload_image_if
mlx/data/Dataset.cpp:425
Functionload_imagenet
Load the ImageNet dataset from the downloaded archives. ImageNet cannot be automatically downloaded so you have to manually download it from
python/mlx/data/datasets/imagenet.py:124
Functionload_imagenet_metadata
Load the metadata for the imagenet classes.
python/mlx/data/datasets/imagenet.py:119
Functionload_images_from_folder
Load images from a folder. For a directory structure like the following .. code-block:: image_folder/ ...class_1/ ....
python/mlx/data/datasets/image_folder.py:8
Functionload_librispeech
Load the librispeech dataset directly from the TAR archive. Args: root (Path or str, optional): The The directory to load/save the data.
python/mlx/data/datasets/librispeech.py:108
Functionload_libritts_r
Load the LibriTTS-R dataset directly from the TAR archive. Args: root (Path or str, optional): The The directory to load/save the data. I
python/mlx/data/datasets/libritts_r.py:98
Functionload_mnist
Load a buffer with the MNIST dataset. If the data doesn't exist download it and save it for the next time. Args: root (Path or str,
python/mlx/data/datasets/mnist.py:70
Methodload_numpy
mlx/data/Dataset.cpp:442
Methodload_numpy_if
mlx/data/Dataset.cpp:452
Functionload_speechcommands
Load the Speech Commands (v0.0.2) [1] dataset directly from the TAR archive. Args: root (Path or str, optional): The The directory to loa
python/mlx/data/datasets/speechcommands.py:88
Methodload_video
mlx/data/Dataset.cpp:467
Methodload_video_if
mlx/data/Dataset.cpp:478
Functionload_wikitext_lines
Fetch the WikiText dataset and load it as a stream of lines. Args: root (Path or str, optional): The The directory to load/save the data.
python/mlx/data/datasets/wikitext.py:35
Methodlog_run
(self, run_name, fn, *args, **kwargs)
benchmarks/utils.py:14
← previousnext →401–500 of 662, ranked by callers