module onnxrt.ops_cpu.op_negative_log_likelihood_loss#

Inheritance diagram of mlprodict.onnxrt.ops_cpu.op_negative_log_likelihood_loss

Short summary#

module mlprodict.onnxrt.ops_cpu.op_negative_log_likelihood_loss

Runtime operator.

source on GitHub

Classes#

class

truncated documentation

NegativeLogLikelihoodLoss

Python runtime for function NegativeLogLikelihoodLoss.

Functions#

function

truncated documentation

_compute_negative_log_likelihood_loss

Modified version of softmaxcrossentropy.py

Properties#

property

truncated documentation

args_default

Returns the list of arguments as well as the list of parameters with the default values (close to the signature). …

args_default_modified

Returns the list of modified parameters.

args_mandatory

Returns the list of optional arguments.

args_optional

Returns the list of optional arguments.

atts_value

Returns all parameters in a dictionary.

Methods#

method

truncated documentation

__init__

_infer_shapes

_infer_sizes

_infer_types

_run

Documentation#

Runtime operator.

source on GitHub

class mlprodict.onnxrt.ops_cpu.op_negative_log_likelihood_loss.NegativeLogLikelihoodLoss(onnx_node, desc=None, **options)#

Bases: OpRun

Python runtime for function NegativeLogLikelihoodLoss.

source on GitHub

__init__(onnx_node, desc=None, **options)#
_infer_shapes(x, target, weight=None)#

Should be overwritten.

source on GitHub

_infer_sizes(*args)#

Should be overwritten.

source on GitHub

_infer_types(x, target, weight=None)#

Should be overwritten.

source on GitHub

_run(x, target, weight=None, attributes=None, verbose=0, fLOG=None)#

Should be overwritten.

source on GitHub

mlprodict.onnxrt.ops_cpu.op_negative_log_likelihood_loss._compute_negative_log_likelihood_loss(x, target, weight=None, reduction=b'mean', ignore_index=None)#

Modified version of softmaxcrossentropy.py to handle other type than float32.

source on GitHub