package.common.basesession Namespace Reference

Data Structures
class	BaseSession

Functions
	num_elems (tensor_info)
	tensorinfo ()

Function Documentation

num_elems()

package.common.basesession.num_elems

(

tensor_info

)

Get the total number of elements in nnfw_tensorinfo.dims.

Definition at line 8 of file basesession.py.

def num_elems(tensor_info):
    """Get the total number of elements in nnfw_tensorinfo.dims."""
    n = 1
    for x in range(tensor_info.rank):
        n *= tensor_info.dims[x]
    return n
 
 

Referenced by package.common.basesession.BaseSession.set_inputs().

tensorinfo()

package.common.basesession.tensorinfo

(

)

Shortcut to create a fresh tensorinfo instance.
Raises:
    OnertError: If the C-API call fails.

Definition at line 181 of file basesession.py.

def tensorinfo():
    """
    Shortcut to create a fresh tensorinfo instance.
    Raises:
        OnertError: If the C-API call fails.
    """
 
    try:
        return infer.nnfw_tensorinfo()
    except OnertError:
        raise
    except Exception as e:
        raise OnertError(f"Failed to create tensorinfo: {e}") from e