Contents

struct

SushiAI::Graph::PoolResult

Holds pooled tensor output and argmax indices.

Declared in
include/SushiAI/graph/builder.hpp

Public attributes

ValueId output

The pooled [N,H_o,W_o,C] value.

ValueId indices

The INT64 argmax, same shape.