Returns the default device id for the device type passed as an argument. For CPU it's always None, for GPU it's the current CUDA dev

The device on which the batch resides (or will reside, in case of lazy evaluation).

The element type of the tensors in the batch.

* @brief Return empty shape of specified dimensionality */

Clears the predicate set efficiently

Enters a conditional section. Conditional sections define an entry node, and one or more branches. Args: section_id: Hasha

Enters a regular section. Regular sections admit exit jumps, which end the section. Args: section_id: Hashable, the same n

Error function. \param arg function argument \return error function value of \a arg template<typename T> typename enable<expr,T>::type erf(T arg) { re

Returns provided pipeline executor statistics metadata as a dictionary. Each key in the dictionary is the operator name. To enable it use ``ex

Exits a conditional section.

Exits a regular section.

* @brief Return shape of specified dimensionality filled with value */

Nearest integer not greater than half value. \param arg half to round \return nearest integer not greater than \a arg template<typename T> typename en

Expects a two dimensional flow image of shape [H,W,2] According to the C++ source code of Daniel Scharstein According to the Matlab sourc

Obtains alias for given version if exists. Otherwise return None Parameters ---------- version : str or PckgVer P

Returns the batch size in samples :param batch: List of input batches, if there is one input a batch can be either a numpy

Computes the center coordinates, height and width of the boxes. Args: scope: name scope of the function. Returns:

Returns device string like 'CPU' or 'GPU'.

* @brief When we have logical condition, the true group is expected to go through the first (0) * output and input (respectively for split and merge)

Gets the mask

Get nested attribute by splitting attr_path on '.' and calling getattr for each part. Example: get_nested_attr(fn, "experimental.med

Obtains the number of available versions for given cuda_version Parameters ---------- cuda_version : str, optional, default =

Gets the number of threads in the default thread pool. The value is determined by (in decreasing priority): 1. The value (not None) pass

Generate section with references for given operator or module

Utility to get ArraySpec for given JAX array.

Normalizes the size parameter. Called once at initialization. Returns the size in a canonical form: - ``int`` — resize the shorter e

* @brief Checks whether the tensors in the list are packed sequentially, without gaps. * * The function checks if the data pointer to a next sam

Classification implementation. \param arg value to classify \retval true if not a number \retval false else

A deprecated method of providing the pipeline with external inputs. This function can be overridden by a user-defined pipeline to per

The semantic layout of the tensor, e.g. HWC, CHW. The layout assigns meaning to the axes. It affects the way in which the data is