onnx_chainer.context.Context

class onnx_chainer.context.Context(model)

Context of converter

This context shares names during exporting.

name_list

list of being exported as ONNX node name with pinned or not, keyed by instance ID. When the target variable is chainer.Variable or chainer.Parameter, instance ID of ndarray held by the variable is also put as key, because some functions like F.where internally unwrap variable.

Type:dict

Methods

add_const(array, name)

Add a constant array as an ONNX Constant node.

Returns:registered name. Return type:str

add_param(array, name, use_original_name=False)

Add a parameter array as an ONNX initializer.

Returns:registered name. Return type:str

get_link(param)

Return link with name which has the param.

Parameters:param (chainer.Parameter) – the target param. Returns:name and link. returns None when not found. Return type:tuple

get_name(variable)

is_pinned(variable)

set_name(variable, name, pinned=False)

Set ONNX node name

Parameters:

• variable (var) – target variable
• name (str) – name to be exported as ONNX node name
• pinned (bool) – if True, the name will not be overwritten in subsequence process.