selector

torch_to_nnef.op.aten.selector

argsort

argsort(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:argsort' to NNEF.

embedding

embedding(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:embedding' to NNEF.

gather

gather(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:gather' to NNEF.

index_

index_(node, op_helper, inference_target, **kwargs)

Translate aten::index to NNEF.

Fragment gather<?>(. input: tensor<?>, # the tensor to gather from indices: tensor, # the indices to gather at axis: integer = 0 ) # the axis to gather at -> ( output: tensor<?> )

torch ir, in this case structure indexes_node with: a list of n values where n <= input_node rank each value is either a constant or a tensor. if the constant is None this means the full dimension

index_select

index_select(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:index_select' to NNEF.

masked_fill

masked_fill(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:masked_fill' to NNEF.

narrow

narrow(node, op_helper, **kwargs)

Fancy slice made in PyTorch.

torch.narrow(input, dim, start, length)

Example:

import torch x = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) torch.narrow(x, 0, 0, 2) tensor([[1, 2, 3], [4, 5, 6]])

scatter

scatter(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:scatter' to NNEF.

select

select(node, op_helper, **kwargs)

Map PyTorch: 'aten:select' to NNEF.

slice_

slice_(node, torch_graph, inference_target, op_helper, **kwargs)

Map PyTorch: 'aten:slice' to NNEF.

sort

sort(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:sort' to NNEF.

topk

topk(node, op_helper, inference_target, **kwargs)

Map PyTorch: 'aten:topk' to NNEF.

tract_pre_0_21_7_slice

tract_pre_0_21_7_slice(node, torch_graph, nnef_spec_strict, has_dynamic_axes, op_helper, **kwargs)

Old version of slice for tract version prior to 0.21.7.

where

where(node, op_helper, **kwargs)

Map PyTorch: 'aten:where' to NNEF.