Searched refs:reduce_axes (Results 1 – 1 of 1) sorted by relevance
3976 const uint32_t reduce_axes = 1 << (num_dims - 1); in ConvertTopK() local3978 *const_cast<nvinfer1::ITensor*>(tensor), op, k, reduce_axes); in ConvertTopK()