matmul

torch_to_nnef.op.aten.matmul

baddbmm

baddbmm(g, node, name_to_tensor, inference_target, **kwargs)

Map PyTorch: 'aten:baddbmm', 'aten:addmm' to NNEF.

einsum

einsum(g, node, name_to_tensor, inference_target, **kwargs)

Map PyTorch: 'aten:einsum' to NNEF.

linear

linear(g, node, name_to_tensor, null_ref, inference_target, **kwargs)

Map PyTorch: 'aten:linear' to NNEF.

matmul

matmul(g, node, name_to_tensor, **kwargs)

Map PyTorch: 'aten:matmul', 'aten:bmm', 'aten:mm' to NNEF.