SymbolicContext¶
-
class
torch.onnx.SymbolicContext(params_dict, env, cur_node, onnx_block)[source]¶ Provides extra context for symbolic functions.
- Parameters
params_dict (Dict[str, _C.IValue]) – Mapping from graph initializer name to IValue.
env (Dict[_C.Value, _C.Value]) – Mapping from Torch domain graph Value to ONNX domain graph Value.
cur_node (_C.Node) – Current node being converted to ONNX domain.
onnx_block (_C.Block) – Current ONNX block that converted nodes are being appended to.
