package.infer Namespace Reference

Data Structures
class	session

Functions
	num_elems (tensor_info)
	tensorinfo ()

Function Documentation

num_elems()

package.infer.num_elems

(

tensor_info

)

Get the total number of elements in nnfw_tensorinfo.dims.

Definition at line 8 of file infer.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.infer.session.set_inputs(), and package.infer.session.set_outputs().

tensorinfo()

package.infer.tensorinfo

(

)

Definition at line 57 of file infer.py.

def tensorinfo():
    return libnnfw_api_pybind.nnfw_tensorinfo()