def _create_npu_activation(serial_activation: spec.SerialActivation) -> vapi.NpuActivation: """This is a helper function to capture a list of arguments to create Vela NpuActivation object.""" if serial_activation.op == "NONE": return None if ( serial_activation.op == "CLIP" and serial_activation.clip_min == 0 and serial_activation.clip_max == 0 ): return None op_map = { "CLIP": vapi.NpuActivationOp.NONE_OR_RELU, "TANH": vapi.NpuActivationOp.TABLE_LOOKUP, "SIGMOID": vapi.NpuActivationOp.TABLE_LOOKUP, "LUT": vapi.NpuActivationOp.TABLE_LOOKUP, } op = str(serial_activation.op.value) assert op in op_map.keys() act_op = vapi.NpuActivation(op_map[op]) if serial_activation.op == "CLIP": act_op.min = int(serial_activation.clip_min.value) act_op.max = int(serial_activation.clip_max.value) if op_map[op] == vapi.NpuActivationOp.TABLE_LOOKUP: act_op.lookup_table_index = 0 return act_op
def _create_npu_activation(serial_activation): """This is a helper function to capture a list of arguments to create Vela NpuActivation object""" if serial_activation.op == "NONE": return None if (serial_activation.op == "CLIP" and serial_activation.clip_min == 0 and serial_activation.clip_max == 0): return None op_map = { "CLIP": vapi.NpuActivationOp.NONE_OR_RELU, "TANH": vapi.NpuActivationOp.TANH, "SIGMOID": vapi.NpuActivationOp.SIGMOID, } op = str(serial_activation.op.value) assert op in op_map.keys() act_op = vapi.NpuActivation(op_map[op]) act_op.min = int(serial_activation.clip_min.value) act_op.max = int(serial_activation.clip_max.value) return act_op