Searched refs:dst_check (Results 1 – 1 of 1) sorted by relevance
261 std::string dst_check; in GetReduceKernelCode() local263 if (!dst_check.empty()) { in GetReduceKernelCode()264 dst_check += " || "; in GetReduceKernelCode()266 dst_check += "DST_" + axis_to_coord[axis] + " >= args.dst_tensor." + in GetReduceKernelCode()269 if (!dst_check.empty()) { in GetReduceKernelCode()270 c += " if (" + dst_check + ") return;\n"; in GetReduceKernelCode()