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