yobx.tensorflow.ops.gather_ops#

Converter for TF GatherV2 op → ONNX Gather.

Indexing / lookup#

GatherV2

yobx.tensorflow.ops.gather_ops.convert_gather_v2(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: Operation) → str[source]#

Converts TF GatherV2 (tf.gather) → ONNX Gather.

TF GatherV2 takes three inputs: params, indices, and a scalar axis constant. The axis value is read from that constant and forwarded as the axis attribute of the ONNX Gather node.