* @brief Computes the maximum size in each dimension * * @param a First size to compare. * @param b Second size to compare. * @return The size wit

* @brief Function to permute the dimensions of a tensor to a new layout. * * This function rearranges the dimensions of the tensor according to a ne

Create a TensorRT engine from an ONNX model. Args: onnx: Path to the ONNX model. output: Path to the output TensorRT engine.

Returns the string representation of the tensor layout * * @param [in] layout Layout to be queried * * @returns Null-terminated string with the la

Recursively gathers values of all keys of input_dict in another dictionary. This is useful for computing various stats on the data such as me

Resets the contents of the callback, performing cleanup, if necessary * * @param function the function to be called, taking the target and

run operator