Searched refs:reduce_axes (Results 1 – 1 of 1) sorted by relevance
5711 const uint32_t reduce_axes = 1 << trt_axis; in ConvertArgMinMax() local5713 *inputs.at(0).tensor(), topk_op, 1, reduce_axes); in ConvertArgMinMax()5765 const uint32_t reduce_axes = 1 << (num_dims - 1); in ConvertTopK() local5767 params->converter->network()->addTopK(*tensor, op, k, reduce_axes); in ConvertTopK()