Create a new NumPy array containing the contents of this device buffer. Returns: np.ndarray: The newly created NumPy arr

Get the logging verbosity for the given path. Args: path (str): The path Returns: int: The logging

Logs a message to stdout with WARNING severity. Args: message (Union[str, Callable() -> str]): A str

Logs a message to stdout with CRITICAL severity and raises an exception. Args: message (Union[str, Callable() -> str]):

Simple helper function that

Adds imports to this script. For example: :: script.add_import("numpy") # Equivalent to: import numpy

Logs a message to stdout with INFO severity. Args: message (Union[str, Callable() -> str]): A string

\brief Get the layer output corresponding to the given index. \return The indexed output tensor, or nullptr if the index is out of range or the tenso

Creates a node, adds it to this graph, and optionally creates its input and output tensors. The input and output lists can include v

Helper function to run the `polygraphy run` command with nececssary args

Logs a message to stdout with ERROR severity. Args: message (Union[str, Callable() -> str]): A strin

Return pointer to IDimensionExp for given value.

\brief Resizes the buffer. This is a no-op if the new size is smaller than or equal to the current capacity.

Convenience function to get a producer node of one of this node's input tensors. Note that the parameters are swapped compared to the

Convenience function to get a consumer node of one of this node's output tensors. For example: :: assert node.o

Creates a string representation that will invoke the specified object, with the specified arguments. Args: type_str (str): A str

Logs a message to stdout with VERBOSE severity. Args: message (Union[str, Callable() -> str]): A str

Creates a read-only DeviceView from this DeviceArray. Args: shape (Sequence[int]): The desired shape

Makes a shallow copy of this node, overriding input and output information. Note: Generally, you should only ever make a copy of a G

Indicates a string is safe to use and will not compromise security. NOTE: The caller is reponsible for checking that the string is actually

Bypass the module

\brief Return the name of the loop. \see setName()

IPluginV2DynamicExt Methods

Adds a loader to the script. If the loader is a duplicate, returns the existing loader instead. Args: loader_str

Adds code to the given script that performs the functionality provided by this argument group. For example, ``TrtConfigArgs`` would

Save this script to the specified destination. Args: dest (file-like): A file-like object that defin

Creates a tensor map of all the tensors used by this graph by walking over all nodes. Empty tensors are omitted from this map. Tenso

Takes user parameters and adds LayerConfig object to a list for each add call. Args: name (Union[str, List]): Name of th

Removes unused nodes and tensors from the graph. A node or tensor is considered unused if it does not contribute to any of the graph

Full test workflow: quantization, obtain expected QDQ node placements, check node placements against expected. Args: test_assets (Cr

\brief Set a single build mode flag. Add the input builder mode flag to the already enabled flags. \see setFlags()

Get the input tensor of a node iff the input tensor is not already marked a graph constant.

\brief Resets the builder state to default values.

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 Returns the host buffer corresponding to tensorName. Returns nullptr if no such tensor can be found.

Folds constants in-place in the graph. The graph must be topologically sorted prior to calling this function (see `toposort()`).

Collect histogram

Decoder step using stored states, attention and memory PARAMS ------ decoder_input: previous mel output RETURNS

\brief Get the layer input corresponding to the given index. \param index The index of the input tensor. \return The input tensor, or nullptr if the

Creates a function that compares two IterationResults, and can be used as the `compare_func` argument in ``Comparator.compare_accurac

Topologically sort the graph in place. Args: recurse_subgraphs (bool): Whether to recursively topolo

Logs a message to stdout with EXTRA_VERBOSE severity. Args: message (Union[str, Callable() -> str]):

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