MCPcopy Create free account

hub / github.com/NVIDIA/DALI / endpoints

Endpoints95 in github.com/NVIDIA/DALI

⨍ Functions 17,951◇ Types & classes 3,242↳ Endpoints 95
Route_call
utils.recompute_grad(self._global_params.grad_checkpoint)
docs/examples/use_cases/tensorflow/efficientdet/model/backbone/efficientnet_model.py:None
Route_call
utils.recompute_grad(self.grad_checkpoint)
docs/examples/use_cases/tensorflow/efficientdet/model/utils/layers.py:None
Route_get_input_device
NVTXRange("get_input_device", category="op_builder")
dali/python/nvidia/dali/experimental/dynamic/_ops.py:None
Route_normalize
adjust_input
dali/python/nvidia/dali/experimental/torchvision/v2/functional/normalize.py:None
Route_pad
adjust_input
dali/python/nvidia/dali/experimental/torchvision/v2/functional/pad.py:None
Route_test_not_a_sequence_layout
raises( RuntimeError, "Per-frame argument input must be a sequence. " "The input layout should start w
dali/test/python/operator_1/test_per_frame.py:None
Route_test_permute_batch_out_of_range
raises( RuntimeError, glob="Sample index out of range. * is not a valid index for an input batch of *
dali/test/python/operator_1/test_batch_permute.py:None
Routecenter_crop
adjust_input
dali/python/nvidia/dali/experimental/torchvision/v2/functional/centercrop.py:None
Routecheck_tf_dataset_wrong_input_type
raises( TypeError, glob="`input_datasets` must be a dictionary that maps input names * to input datase
dali/test/python/test_dali_tf_dataset_eager.py:None
Routecond_flip
dax.fn.jax_function(num_outputs=2)
dali/test/python/jax_plugin/test_jax_operator.py:None
Routecrop
adjust_input
dali/python/nvidia/dali/experimental/torchvision/v2/functional/crop.py:None
Routegaussian_blur
adjust_input
dali/python/nvidia/dali/experimental/torchvision/v2/functional/gaussian_blur.py:None
Routeget_compiled_result
_nvtx_range("Getting compiled result")
dali/python/nvidia/dali/experimental/dynamic/_compile.py:None
Routehorizontal_flip
adjust_input
dali/python/nvidia/dali/experimental/torchvision/v2/functional/flips.py:None
Routeimage_slice_decoder_pipe
pipeline_def( batch_size=47, device_id=0, num_threads=ndd.get_num_threads(), prefetch_queue_depth=1
dali/test/python/ndd_vs_fn/test_ndd_vs_fn_decoders.py:None
Routeiterator_fn
decorator(output_map=["images"], reader_name="Reader")
dali/test/python/checkpointing/test_dali_checkpointing_fw_iterators.py:None
Routeiterator_function
peekable_data_iterator(output_map=["data"], reader_name="reader")
dali/test/python/jax_plugin/test_peekable_iterator_decorator.py:None
Routeiterator_function
data_iterator( output_map=output_map, sharding=sharding, last_batch_policy=LastBatchPo
dali/test/python/jax_plugin/test_multigpu.py:None
Routeiterator_function
data_iterator(output_map=["data"], reader_name="reader")
dali/test/python/jax_plugin/test_iterator_decorator.py:None
Routeiterator_function
dax.data_iterator( output_map=["data"], sharding=sharding, )
dali/test/python/jax_plugin/jax_op_multi_gpu_sharding.py:None
Routeiterator_function
data_iterator( output_map=["images"], sharding=sharding, )
docs/examples/custom_operations/jax_operator_multi_gpu_process_1.py:None
Routepipe
pipeline_def( batch_size=max_batch_size, device_id=0, num_threads=ndd.get_num_threads(
dali/test/python/ndd_vs_fn/ndd_vs_fn_test_utils.py:None
Routepipeline
pipeline_def(batch_size=len(input_batch), num_threads=4, device_id=0)
dali/test/python/test_dltensor_operator.py:None
Routepipeline
pipeline_def(batch_size=batch_size, num_threads=ndd.get_num_threads(), device_id=0)
dali/test/python/ndd_vs_fn/test_ndd_vs_fn_readers.py:None
Routepipeline
pipeline_def( batch_size=MAX_BATCH_SIZE, num_threads=ndd.get_num_threads(), device_id=
dali/test/python/ndd_vs_fn/test_ndd_vs_fn_noinput.py:None
Routepipeline
pipeline_def( batch_size=MAX_BATCH_SIZE, device_id=0, num_threads=ndd.get_num_threads(
dali/test/python/ndd_vs_fn/test_ndd_vs_fn.py:None
Routeposterize
augmentation(mag_range=(0, 4), mag_to_param=poster_mask_uint8, param_device="auto")
dali/python/nvidia/dali/auto_aug/augmentations.py:None
Routereshape
dax.fn.jax_function(output_layouts=["HWC", "FHWC"], num_outputs=2)
dali/test/python/jax_plugin/test_jax_operator.py:None
Routeresize
adjust_input
dali/python/nvidia/dali/experimental/torchvision/v2/functional/resize.py:None
Routeresized_crop
adjust_input
dali/python/nvidia/dali/experimental/torchvision/v2/functional/crop.py:None
Routergb_to_grayscale
adjust_input
dali/python/nvidia/dali/experimental/torchvision/v2/functional/color.py:None
Routeslow_test_bitnot
cartesian_params(unary_input_kinds, bitnot_operations, integer_types)
dali/test/python/operator_1/test_arithmetic_ops.py:None
Routeslow_test_bitwise_ops
cartesian_params(bin_input_kinds, bitwise_operations, iter_pow(integer_types, 2))
dali/test/python/operator_1/test_arithmetic_ops.py:None
Routeslow_test_mod2
cartesian_params(bin_input_kinds, iter_pow(set(integer_types) - set([np.bool_]), 2))
dali/test/python/operator_1/test_arithmetic_ops.py:None
Routeslow_test_shift_ops
cartesian_params( bin_input_kinds, shift_operations, iter_pow(set(integer_types) - set([np.bool_]), 2) )
dali/test/python/operator_1/test_arithmetic_ops.py:None
Routetest_axis_name_no_initial_layout_multi_dim
raises(RuntimeError, glob="Input layout mismatch")
dali/test/python/operator_1/test_one_hot.py:None
Routetest_bitnot_selected
cartesian_params(unary_input_kinds, bitnot_operations, integer_types)
dali/test/python/operator_1/test_arithmetic_ops.py:None
Routetest_bitwise_ops_selected
cartesian_params( selected_bin_input_kinds, bitwise_operations, iter_pow([np.bool_] + selected_integer_typ
dali/test/python/operator_1/test_arithmetic_ops.py:None
Routetest_choice_validation
params( ( (1,), {"p": 1.5}, "Probabilities must be in range *, but got: 1.5 for sa
dali/test/python/operator_2/test_random_choice.py:None
Routetest_chunks
restrict_platform(min_compute_cap=6.0)
dali/test/python/operator_1/test_inflate.py:None
Routetest_clashing_args
raises( RuntimeError, glob="*Matrix input and `matrix` argument should not be provided at the same tim
dali/test/python/operator_2/test_warp_perspective.py:None
Routetest_consecutive
params(*itertools.product(["cpu", "gpu"], input_gens, pred_gens, pred_gens))
dali/test/python/conditionals/test_pipeline_conditionals.py:None
Routetest_device_change
raises(RuntimeError, glob="Input * for operator * is on * but was on * when created.")
dali/test/python/test_pipeline_debug.py:None
Routetest_device_match_mixed_operator
cartesian_params( [ None, ] + list(range(_backend.GetCUDADeviceCount())), )
dali/test/python/experimental_mode/test_eval_context.py:None
Routetest_eval_context_evaluate_gpu_expr
cartesian_params( [ None, ] + list(range(_backend.GetCUDADeviceCount())), )
dali/test/python/experimental_mode/test_eval_context.py:None
Routetest_fail_per_frame_no_frames
raises( RuntimeError, "Tensor input for argument window_size is specified per frame " "(got F layo
dali/test/python/operator_1/test_gaussian_blur.py:None
Routetest_feed_input_with_source
raises( RuntimeError, glob="Cannot use `feed_input` on the external source 'input' with a `source`"
dali/test/python/test_external_source_multiple_sources.py:None
Routetest_get_property_stateless
stateless_signed_off("get_property")
dali/test/python/checkpointing/test_dali_stateless_operators.py:None
Routetest_imread
cartesian_params( ("cpu", "gpu"), ({}, {"output_type": types.GRAY}), ( images[0],
dali/test/python/experimental_mode/test_imread.py:None
Routetest_incorrect_input_sets
raises( ValueError, glob=( "All argument lists for Multiple Input Sets used with operator"
dali/test/python/test_pipeline_debug.py:None
Routetest_incorrect_variable_batch_size_inside_es
raises(RuntimeError, glob="External source must return outputs with consistent batch size.*")
dali/test/python/test_pipeline_debug.py:None
Routetest_inflate
restrict_platform(min_compute_cap=6.0)
dali/test/python/test_dali_variable_batch_size.py:None
Routetest_inflate_stateless
restrict_platform(min_compute_cap=6.0)
dali/test/python/checkpointing/test_dali_stateless_operators.py:None
Routetest_input_not_provided
raises( ValueError, glob="Found External Source nodes in the Pipeline, that were not assigned
dali/test/python/test_dali_tf_dataset_eager.py:None
Routetest_inputs_batch_change
raises(RuntimeError, glob="Input * for operator * is")
dali/test/python/test_pipeline_debug.py:None
Routetest_inputs_len_change
raises( RuntimeError, glob=("Trying to use operator * with different number of inputs than when" " it
dali/test/python/test_pipeline_debug.py:None
Routetest_invalid_layouts_arg
raises(RuntimeError, "*length of*output_layouts*greater than*")
dali/test/python/operator_2/test_python_function.py:None
Routetest_jax_peekable_iterator_with_variable_shapes_pipeline
raises(ValueError, glob="The shape or type of the output changed between iterations.")
dali/test/python/jax_plugin/test_peekable_iterator.py:None
Routetest_keras_wrong_placement_cpu
raises(tf.errors.OpError, "TF device and DALI device mismatch. TF*: GPU, DALI*: CPU for output")
dali/test/python/test_dali_tf_dataset_mnist_graph.py:None
Routetest_keras_wrong_placement_gpu
raises(tf.errors.OpError, "TF device and DALI device mismatch. TF*: CPU, DALI*: GPU for output")
dali/test/python/test_dali_tf_dataset_mnist_graph.py:None
Routetest_mixed_size_pipeline
raises( tf.errors.FailedPreconditionError, glob="Batch output at index * from DALI pipeline is not uni
dali/test/python/test_dali_tf_dataset_eager.py:None
Routetest_mod_mixed_signedness
cartesian_params(selected_bin_input_kinds, mixed_sign_types)
dali/test/python/operator_1/test_arithmetic_ops.py:None
Routetest_mod_selected
cartesian_params(selected_bin_input_kinds, iter_pow(selected_integer_types, 2))
dali/test/python/operator_1/test_arithmetic_ops.py:None
Routetest_multiple_nests
params(*itertools.product(["cpu", "gpu"], input_gens, num_gens))
dali/test/python/conditionals/test_pipeline_conditionals.py:None
Routetest_nested
params(*itertools.product(["cpu", "gpu"], input_gens, pred_gens, pred_gens))
dali/test/python/conditionals/test_pipeline_conditionals.py:None
Routetest_nested_with_assignment
params(*itertools.product(["cpu", "gpu"], input_gens, pred_gens, pred_gens))
dali/test/python/conditionals/test_pipeline_conditionals.py:None
Routetest_no_input_operators
params(*no_input_ops_test_configuration)
dali/test/python/ndd_vs_fn/test_ndd_vs_fn_noinput.py:None
Routetest_numpy_reader_roi
params(*list(_get_roi_suite_params()))
dali/test/python/reader/test_numpy.py:None
Routetest_numpy_reader_roi_empty_axes
params(*list(_get_roi_empty_axes_params()))
dali/test/python/reader/test_numpy.py:None
Routetest_numpy_reader_roi_error
params(*list(_get_roi_error_params()))
dali/test/python/reader/test_numpy.py:None
Routetest_output_dtype_both_error
raises(TypeError, glob="unexpected*output_dtype*dtype")
dali/test/python/test_pipeline.py:None
Routetest_python_operator_with_input_sets
raises(TypeError, "do not support multiple input sets")
dali/test/python/operator_2/test_python_function.py:None
Routetest_reading_compressed
params(*enum_product(get_dtypes(True), [1, 2, 3], ["cpu", "gpu"]))
dali/test/python/reader/test_fits.py:None
Routetest_reading_uncompressed
params(*enum_product(get_dtypes(False), [1, 2, 3, 4], ["cpu", "gpu"]))
dali/test/python/reader/test_fits.py:None
Routetest_sample_inflate
restrict_platform(min_compute_cap=6.0)
dali/test/python/operator_1/test_inflate.py:None
Routetest_scalar_shape
restrict_platform(min_compute_cap=6.0)
dali/test/python/operator_1/test_inflate.py:None
Routetest_shift_ops_selected
cartesian_params(selected_bin_input_kinds, shift_operations, iter_pow(selected_integer_types, 2))
dali/test/python/operator_1/test_arithmetic_ops.py:None
Routetest_tf_dataset_es_num_outputs_provided
raises( ValueError, glob="The parameter ``num_outputs`` is only valid when using ``source`` to
dali/test/python/test_dali_tf_dataset_eager.py:None
Routetest_tf_dataset_multi_input
params(*_generate_tf_dataset_multi_input_test_cases())
dali/test/python/test_dali_tf_dataset_eager.py:None
Routetest_tf_dataset_multi_input
params(*_generate_tf_dataset_multi_input_test_cases())
dali/test/python/test_dali_tf_dataset_graph.py:None
Routetest_tf_dataset_with_constant_input
params(*_generate_tf_dataset_with_constant_input_test_cases())
dali/test/python/test_dali_tf_dataset_eager.py:None
Routetest_tf_dataset_with_constant_input
params(*_generate_tf_dataset_with_constant_input_test_cases())
dali/test/python/test_dali_tf_dataset_graph.py:None
Routetest_tf_dataset_with_random_input
params(*_tf_dataset_with_random_input_test_cases)
dali/test/python/test_dali_tf_dataset_eager.py:None
Routetest_tf_dataset_with_random_input_gpu
params(*_tf_dataset_with_random_input_gpu_test_cases)
dali/test/python/test_dali_tf_dataset_eager.py:None
Routetest_tf_experimental_inputs_disabled
raises(TypeError, glob="Dataset inputs are allowed only in *DALIDatasetWithInputs")
dali/test/python/test_dali_tf_dataset_eager.py:None
Routetest_total_no_chunks
restrict_platform(min_compute_cap=6.0)
dali/test/python/operator_1/test_inflate.py:None
Routetest_types_and_shapes
params(*list(_get_type_and_shape_params()))
dali/test/python/reader/test_numpy.py:None
Routetest_unsupported_types
params(*list(_get_unsupported_param()))
dali/test/python/reader/test_numpy.py:None
Routetest_validation
restrict_platform(min_compute_cap=6.0)
dali/test/python/operator_1/test_inflate.py:None
Routetest_variable_batch_size
raises(RuntimeError, glob="Batch size must be uniform across an iteration. Input*")
dali/test/python/test_pipeline_debug.py:None
Routetest_wrong_out_grid_size
raises(RuntimeError, "Output grid size: 3 is not supported, supported are:")
dali/test/python/operator_1/test_optical_flow.py:None
Routetest_wrong_outputs_number
raises(Exception, glob="Unexpected number of outputs*got 3*expected 2")
dali/test/python/operator_2/test_python_function.py:None
Routetest_wrong_outputs_number_dl
raises(Exception, glob="Unexpected number of outputs*got 3*expected 2")
dali/test/python/operator_2/test_python_function.py:None
Routeto_grayscale
adjust_input
dali/python/nvidia/dali/experimental/torchvision/v2/functional/color.py:None
Routevertical_flip
adjust_input
dali/python/nvidia/dali/experimental/torchvision/v2/functional/flips.py:None