MNN.OpInfo

class OpInfo

OpInfo是用来描述推理过程中一个执行层的具体信息的数据类型,其中包含了该层的Op类型,名称,计算量(flops)

用法参考runSessionWithCallBackInfo


OpInfo()

创建一个空OpInfo

在实际使用中创建空OpInfo没有意义,仅在runSessionWithCallBackInfo函数中用作返回信息

参数:

  • None

返回:OpInfo对象

返回类型:OpInfo


getName()

获取该层的名称

参数:

  • None

返回:该层的名称

返回类型:str


getType()

获取该层的算子类型

参数:

  • None

返回:该层的算子类型

返回类型:str


getFlops()

获取该层的计算量

参数:

  • None

返回:该层的计算量

返回类型:float


Example

import MNN

interpreter = MNN.Interpreter("mobilenet_v1.mnn")
session = interpreter.createSession()
# set input
# ...
def begin_callback(tensors, opinfo):
    print('layer name = ', opinfo.getName())
    print('layer op = ', opinfo.getType())
    # print('layer flops = ', opinfo.getFlops())
    for tensor in tensors:  
        print(tensor.getShape())
    return True
def end_callback(tensors, opinfo):
    print('layer name = ', opinfo.getName())
    print('layer op = ', opinfo.getType())
    # print('layer flops = ', opinfo.getFlops())
    for tensor in tensors:  
        print(tensor.getShape())
    return True
interpreter.runSessionWithCallBackInfo(session, begin_callback, end_callback)