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 6 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(), and package.common.basesession.BaseSession.set_outputs().

tensorinfo()

package.common.basesession.tensorinfo

(

)

Definition at line 99 of file basesession.py.

def tensorinfo():
    return libnnfw_api_pybind.infer.nnfw_tensorinfo()