r"""Get weight's quantization dtype as the method from ``qconfig``.

! * \brief whether var value is constant */

whether memory comes from owner_var->dev_tensor()

Natural logarithm of gamma function. \param arg function argument \return natural logarith of gamma function for \a arg template<typename T> typename

r"""Calculate and print ``model``'s statistics by adding hook and record Module's inputs outputs size. Args: model: model that need to ge

total number of devices within the clique

get offset in number of logical elements in the layout

r"""Element-wise power. Calculates an implementation-dependent approximation of exponentiation by raising each element :math:`x_i` (the

set the callback in async model

update shape

whether tensor shape is valid (i.e. ndim != 0)

initialize a callback-based context

initialize a memory-decode context

For perf reasons this code is duplicated in stbir__resample_horizontal_upsample/downsample, if you change it here change it there too.

================== ProfilerBase =================*/

r"""Returns whether tensor.shape returns a tensor instead of a tuple

! * \brief get value if this is valid; otherwise returns default value * * Note: this function returns by value rather than by referenc

get current verbosity setting