o e/ã@s*ddlmZddlmZGdd„dƒZdS)é)Ú service_pb2é)Ú raise_errorc@s<eZdZdZddd„Zdd„Zddd„Zd d „Zd d „Zd S)ÚInferRequestedOutputa”An object of InferRequestedOutput class is used to describe a requested output tensor for an inference request. Parameters ---------- name : str The name of output tensor to associate with this object class_count : int The number of classifications to be requested. The default value is 0 which means the classification results are not requested. rcCs4t ¡ ¡|_||j_|dkr||jjd_dSdS)NrÚclassification)rZModelInferRequestZInferRequestedOutputTensorÚ_outputÚnameÚ parametersÚ int64_param)ÚselfrZ class_count©r úSD:\Projects\ConvertPro\env\Lib\site-packages\tritonclient/grpc/_requested_output.pyÚ__init__/s ÿzInferRequestedOutput.__init__cCs|jjS)zˆGet the name of output associated with this object. Returns ------- str The name of output )rr©r r r r r5szInferRequestedOutput.namecCsNd|jjvr tdƒ||jjd_||jjd_|dkr%||jjd_dSdS)aTMarks the output to return the inference result in specified shared memory region. Parameters ---------- region_name : str The name of the shared memory region to hold tensor data. byte_size : int The size of the shared memory region to hold tensor data. offset : int The offset, in bytes, into the region where the data for the tensor starts. The default value is 0. Raises ------ InferenceServerException If failed to set shared memory for the tensor. rz3shared memory can't be set on classification outputÚshared_memory_regionÚshared_memory_byte_sizerÚshared_memory_offsetN)rr rZ string_paramr )r Z region_nameZ byte_sizeÚoffsetr r r Úset_shared_memory?s ÿz&InferRequestedOutput.set_shared_memorycCs4|jj dd¡|jj dd¡|jj dd¡dS)zåClears the shared memory option set by the last call to InferRequestedOutput.set_shared_memory(). After call to this function requested output will no longer be returned in a shared memory region. rNrr)rr Úpoprr r r Úunset_shared_memoryZsz(InferRequestedOutput.unset_shared_memorycCs|jS)zÅRetrieve the underlying InferRequestedOutputTensor message. Returns ------- protobuf message The underlying InferRequestedOutputTensor protobuf message. )rrr r r Ú _get_tensoresz InferRequestedOutput._get_tensorN)r) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrrr r r r r!s   rN)Ztritonclient.grpcrÚ_utilsrrr r r r Ús