MCPcopy Create free account

hub / github.com/NVIDIA/DALI / functions

Functions17,951 in github.com/NVIDIA/DALI

↓ 21 callersFunctionreshape
(image)
dali/test/python/jax_plugin/test_jax_operator.py:322
↓ 21 callersFunctionskip_if_m60
Skip the test if the GPU is M60. The video decoder is not supported in full on M60.
dali/test/python/test_utils.py:87
↓ 20 callersMethodCopy
dali/pipeline/operator/builtin/copy.h:31
↓ 20 callersMethodGet
dali/operators/util/axis_args.cc:83
↓ 20 callersMethodGet
dali/pipeline/operator/builtin/input_operator.h:49
↓ 20 callersMethodGetSubexpressionCount
dali/operators/math/expressions/expression_tree.h:111
↓ 20 callersMethodRunExe
dali/pipeline/operator/builtin/external_source_test.cc:353
↓ 20 callersMethodRunOutputs
dali/pipeline/operator/builtin/external_source_test.cc:357
↓ 20 callersMethodSetSample
dali/pipeline/data/tensor_list.cc:273
↓ 20 callersMethodappend
include/dali/core/bitmask.h:248
↓ 20 callersFunctioncapture_processes
Need to be called to register the processes created by the pool. It is later used by the teardown_function to check if no process stayed alive aft
dali/test/python/test_pool_utils.py:20
↓ 20 callersMethodclone
Create a new RNG with the same state Returns ------- RNG A new RNG which will continue the same sequence as this
dali/python/nvidia/dali/experimental/dynamic/random.py:88
↓ 20 callersFunctionclose
(a, b)
dali/test/python/operator_2/test_random_resized_crop.py:22
↓ 20 callersMethodget
Convenience function for accessing box coordinates. Returns: a tensor with shape [N, 4] representing box coordinates.
docs/examples/use_cases/tensorflow/efficientdet/pipeline/anchors_utils/box_list.py:100
↓ 20 callersFunctionisnan
Check for NaN. \param arg number to check \retval true for NaNs \retval false else template<typename T> typename enable<bool,T>::type isnan(T arg) { r
include/dali/util/half.hpp:2775
↓ 20 callersMethodregister_metric
(self, metric_name, meter, verbosity=0, metadata={})
docs/examples/use_cases/pytorch/efficientnet/image_classification/logger.py:228
↓ 20 callersMethodscale
(self, wc, dc, dis, divisor=8)
docs/examples/use_cases/pytorch/efficientnet/image_classification/models/efficientnet.py:99
↓ 20 callersMethodserialize
Serialize the pipeline definition to a Protobuf string. .. note:: This function doesn't serialize the pipeline's internal state.
dali/python/nvidia/dali/pipeline.py:1698
↓ 20 callersFunctionsetup_function
Prepare for the check if all started processes are no longer children of current process
dali/test/python/test_pool_utils.py:32
↓ 20 callersFunctionteardown_function
Check if there are no children processes started by the test after it ended. Be sure to call `capture_processes` in the test.
dali/test/python/test_pool_utils.py:42
↓ 19 callersMethodBuild
dali/pipeline/graph/op_graph2.cc:265
↓ 19 callersMethodOutputs
dali/pipeline/executor/executor_impl.h:538
↓ 19 callersMethodSeekRead
dali/util/s3_file.cc:41
↓ 19 callersMethodSetInput
dali/operators/imgcodec/util/convert_gpu_test.cc:95
↓ 19 callersFunctionSimplifyShapesForBroadcasting
dali/operators/math/expressions/broadcasting.cc:252
↓ 19 callersMethodStart
dali/core/nvtx.cc:31
↓ 19 callersFunctionasnumpy
(tensor)
dali/test/python/experimental_mode/test_tensor.py:24
↓ 19 callersMethodassertAstMatches
(self, actual_node, expected_node_src, expr=True)
dali/test/python/autograph/pyct/test_parser.py:31
↓ 19 callersMethodclear
include/dali/core/mm/detail/free_list.h:88
↓ 19 callersFunctioncuInitChecked
dali/core/dynlink_cuda.cc:52
↓ 19 callersMethodevent
dali/pipeline/executor/executor2/exec_node_task.h:67
↓ 19 callersMethodget
Args: ---------- num_samples : optional positive integer Maximal number of messages to take from the queue, if se
dali/python/nvidia/dali/_multiproc/shared_queue.py:154
↓ 19 callersFunctionmake_test_tensor
(shape=(1, 3, 10, 10))
dali/test/python/torchvision/test_tv_totensor.py:45
↓ 19 callersMethodset_sample_dim
dali/pipeline/data/tensor_list.cc:427
↓ 19 callersMethodslice
dali/kernels/imgproc/structure/connected_components.h:56
↓ 19 callersMethodvisit
(self, node)
dali/python/nvidia/dali/_autograph/pyct/static_analysis/type_inference.py:220
↓ 19 callersMethodwait
dali/core/access_order.cc:35
↓ 19 callersFunctionwebdataset_raw_pipeline
( paths, index_paths, ext, case_sensitive_extensions=True, missing_component_behavior="emp
dali/test/python/webdataset_base.py:28
↓ 18 callersMethodAddOperator
dali/pipeline/operator/checkpointing/checkpoint.cc:30
↓ 18 callersMethodBuild
dali/pipeline/pipeline.cc:557
↓ 18 callersFunctionLoadFromFile
dali/operators/reader/loader/coco_loader.cc:157
↓ 18 callersFunctionSaveToFile
dali/operators/reader/loader/coco_loader.cc:97
↓ 18 callersFunctionSetInputSourceInfo
dali/pipeline/executor/source_info_propagation_test.cc:38
↓ 18 callersMethodSetSkipSample
dali/pipeline/data/tensor_list.cc:687
↓ 18 callersMethodSucceed
Adds a precondition. * * Calling Succeed adds the waitable `w` to the task's list of preconditions. * Duplicates are detected and ignored.
include/dali/core/exec/tasking/task.h:460
↓ 18 callersMethodTest
dali/operators/imgcodec/util/convert_test.cc:237
↓ 18 callersMethodTestUnfolding
dali/pipeline/operator/sequence_shape_test.cc:165
↓ 18 callersMethod_get_magnitudes
(self, num_magnitude_bins)
dali/python/nvidia/dali/auto_aug/core/_augmentation.py:327
↓ 18 callersFunctioncat
(images)
dali/test/python/jax_plugin/test_jax_operator.py:410
↓ 18 callersFunctioncompare_against_baseline
( dali_aug, baseline_op, get_data, batch_size=default_batch_size, dev="gpu", eps=1e-7,
dali/test/python/auto_aug/test_augmentations.py:57
↓ 18 callersFunctiondaliPipelineBuild
dali/c_api_2/pipeline.cc:322
↓ 18 callersFunctiondali_pipeline
()
dali/test/python/jax_plugin/utils.py:40
↓ 18 callersMethoddeallocate_async
! * \brief Deallocates the storage pointed to by `__p` in stream order on * `__stream`. * * `__p` must have been returned by a prior call
include/dali/core/mm/cuda_memory_resource.h:588
↓ 18 callersMethoddevice
(self)
dali/python/nvidia/dali/ops/__init__.py:627
↓ 18 callersMethoddim
dali/kernels/reduce/reduce_cpu.h:76
↓ 18 callersMethodend
dali/operators/video/reader/video_reader_decoder_op.cc:68
↓ 18 callersMethodexp
Exponential implementation. \param arg function argument \return function value stored in single-preicision
include/dali/util/half.hpp:1358
↓ 18 callersFunctiongather_ids
(pipes, epochs_run=0, batch_size=1, num_gpus_arg=None, gpus_arg=None)
dali/test/python/reader/test_shuffling.py:174
↓ 18 callersFunctiongeneric_execute
Given a Python `function` (taking some positional arguments) and a list of sample generators, execute the function twice on batches of data genera
dali/test/python/conditionals/test_pipeline_conditionals.py:114
↓ 18 callersMethodhas_stream
* @brief Returns true if this workspace has CUDA stream available */
dali/pipeline/workspace/workspace.h:360
↓ 18 callersFunctionpil_baseline
(pil_op)
dali/test/python/auto_aug/test_augmentations.py:45
↓ 18 callersFunctionpipeline
()
dali/test/python/checkpointing/test_dali_checkpointing.py:85
↓ 18 callersMethodpop_back
include/dali/core/bitmask.h:180
↓ 18 callersMethodresult
dali/npp/npp.h:351
↓ 18 callersFunctionrun_pipeline
Verifies, if given pipeline supports iter-to-iter variable batch size This function verifies only if given pipeline runs without crashing.
dali/test/python/test_dali_variable_batch_size.py:168
↓ 17 callersMethodAddTask
dali/core/exec/thread_pool_base.cc:230
↓ 17 callersMethodEnableCheckpointing
dali/pipeline/executor/executor2/exec2.cc:506
↓ 17 callersMethodGetTypeId
dali/operators/math/expressions/expression_tree.h:99
↓ 17 callersMethodLower
dali/pipeline/executor/executor2/exec_graph_lowering.cc:25
↓ 17 callersFunctionParseExpressionString
dali/operators/math/expressions/expression_tree.cc:188
↓ 17 callersFunctionSequentialFill
dali/test/tensor_test_utils.h:398
↓ 17 callersFunctionSetLastError
dali/c_api_2/error_handling.cc:81
↓ 17 callersMethodSize
dali/util/s3_file.cc:65
↓ 17 callersMethod_process_basic_statement
(self, node)
dali/python/nvidia/dali/_autograph/pyct/cfg.py:681
↓ 17 callersFunctionalign_ptr
include/dali/core/mm/detail/align.h:40
↓ 17 callersMethodassertHasDefinedIn
(self, node, expected)
dali/test/python/autograph/pyct/static_analysis/test_reaching_definitions.py:57
↓ 17 callersFunctioncodecpar
dali/operators/video/legacy/reader/video_loader.cc:70
↓ 17 callersFunctioncontiguous_raw_data
* @brief Return an un-typed const pointer to the underlying storage. * The TensorList must be either empty or have a valid type and be contiguous.
dali/pipeline/data/tensor_list.h:869
↓ 17 callersFunctiondaliDeserializeDefault
dali/c_api/c_api.cc:297
↓ 17 callersFunctiondaliGetLastErrorMessage
dali/c_api_2/error_handling.cc:184
↓ 17 callersMethodencode
(self, bboxes_in, labels_in, criteria = 0.5)
docs/examples/use_cases/pytorch/single_stage_detector/src/utils.py:86
↓ 17 callersMethodevaluate
Evaluates the underlying lazy expression, if any. If the batch is a result of a lazy evaluation, calling `evaluate` will cause the e
dali/python/nvidia/dali/experimental/dynamic/_batch.py:768
↓ 17 callersFunctiongenerate_data
Generates an epoch of data, that will be used for variable batch size verification. :param max_batch_size: Actual sizes of every batch in th
dali/test/python/ndd_vs_fn/ndd_vs_fn_test_utils.py:110
↓ 17 callersFunctionget_videos
()
dali/test/python/auto_aug/test_augmentations.py:124
↓ 17 callersFunctionimport_numpy
()
dali/test/python/test_utils.py:113
↓ 17 callersFunctionindex_pipe
(data_source, indexing_func)
dali/test/python/operator_2/test_subscript.py:24
↓ 17 callersMethodinsert
include/dali/core/small_vector.h:494
↓ 17 callersFunctionis_aligned
(sh, align, axes)
dali/test/python/operator_1/test_pad.py:196
↓ 17 callersFunctionmax
Computes maxima of corresponding values in `left` and `right`. :rtype: Tensor or Batch of the type that is calculated based on the type promotion
dali/python/nvidia/dali/experimental/dynamic/math.py:232
↓ 17 callersFunctionreduce
dali/kernels/reduce/reduce_cpu.h:82
↓ 17 callersFunctionsignbit
Check sign. \param arg number to check \retval true for negative number \retval false for positive number template<typename T> typename enable<bool,T>
include/dali/util/half.hpp:2791
↓ 17 callersMethodsource_info
Returns the "source_info" property. Equivalent to self.meta("source_info").
dali/python/nvidia/dali/data_node.py:355
↓ 17 callersMethodto_device
Returns the data batch on the specified device. If the batch already resides on the device specified, the function will return `self
dali/python/nvidia/dali/experimental/dynamic/_batch.py:571
↓ 16 callersMethodAcquire
* @brief Acquires argument data, enforcing that the shape of the data matches the * expected shape or it is a scalar, which can also be broa
dali/pipeline/operator/arg_helper.h:147
↓ 16 callersFunctionAtScopeExit
include/dali/core/call_at_exit.h:39
↓ 16 callersFunctionChannels
* @brief Number of channels in a video */
dali/operators/video/frames_decoder_base.h:224
↓ 16 callersMethodCoinFlip
dali/operators/random/coin_flip.h:53
↓ 16 callersMethodGet
dali/operators/decoder/cache/image_cache_blob.cc:88
↓ 16 callersMethodIsDefined
dali/operators/generic/slice/subscript.h:58
↓ 16 callersMethodRun
dali/kernels/imgproc/convolution/laplacian_cpu.h:74
← previousnext →401–500 of 17,951, ranked by callers