Creates new SharedMem instance that points to already allocated shared memory chunk accessible via provided shared memory ``handle``.

Floor implementation. \param arg value to round \return rounded value

Verifies, if given pipeline supports iter-to-iter variable batch size This function conducts qualitative verification. It compares the resul

crops in middle of mask and image corners.

* @brief Actual size, in bytes, of the allocation. */

Parses a type name into a :py:class:`DType`.

* @brief true if there is a value available (explicit or default) */

* @brief Returns the size of the dimension at the given index */

Computes floor of values in `input`. :rtype: TensorList of floor(input). If input is an integer, the result will be float, otherwise

Constructs a specification for an op with the given schema name. */

A more powerful version of generic_visit for statement blocks. An example of a block is the body of an if statement. This function a

Nearest integer. \param arg half to round \return nearest integer, rounded away from zero in half-way cases template<typename T> typename enable<half,

* @brief Get the dimensionality of the sample in the batch */

Returns the device on top of the thread-local device stack. If the stack is empty, returns the default GPU for the calling thread.

* Returns batch size that the Operator is expected to produce on a given output */

Subscribes to other task's output value. * * This function does two things: * - adds a dependency (Succeed) on the producer task * - creat

Creates a :class:`Device` object form a DLPack device descriptor.

Main entry point for index.py file that replaces a standard index.rst file. The contents of this doc will be used to generate corresponding index

* @brief Remove a block from the list and place it in unused_blocks_ * * @param ppblock pointer to a block pointer - this automatically updates

Tests whether the CFG contains the specified statement edges.

Generic detection of custom user-defined type used for all overloads. Parameters ---------- object Custom object

Returns a pointer

Utility to copy between backends

* @brief Removes given memory range from the list, if present * * This function checks if the list contains given address range and if it does,

Raw CUDA stream pointer, stored as uint64.

Square root implementation. \param arg function argument \return function value stored in single-preicision

Computes ceil of values in `input`. :rtype: TensorList of ceil(input). If input is an integer, the result will be float, otherwise th

See control_flow.SymbolNamer.new_symbol.

Run fn vs ndd operator comparison over an epoch of batches and assert outputs match. For each batch in input_epoch: builds/uses a pipeline that r