Searched defs:reduce_axes (Results 1 – 1 of 1) sorted by relevance
5711 const uint32_t reduce_axes = 1 << trt_axis; in ConvertArgMinMax() local5765 const uint32_t reduce_axes = 1 << (num_dims - 1); in ConvertTopK() local