Searched defs:reduce_axes (Results 1 – 2 of 2) sorted by relevance
322 std::vector<int64_t> reduce_axes; in Compile() local
4826 const uint32_t reduce_axes = 1 << trt_axis; in ConvertArgMinMax() local4882 const uint32_t reduce_axes = 1 << (num_dims - 1); in ConvertTopK() local