onert.common.basesession Namespace Reference

Data Structures
class	BaseSession

Functions
	num_elems (tensor_info)

Function Documentation

num_elems()

onert.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 onert.common.basesession.BaseSession.set_inputs().