Searched refs:dst_coordinates (Results 1 – 1 of 1) sorted by relevance
407 std::string dst_coordinates; in GetReduceKernelCode() local410 if (!dst_coordinates.empty()) { in GetReduceKernelCode()411 dst_coordinates += ", "; in GetReduceKernelCode()414 dst_coordinates += "DST_" + axis_to_coord[a]; in GetReduceKernelCode()416 dst_coordinates += "0"; in GetReduceKernelCode()420 c += " args.dst_tensor.Write(result, " + dst_coordinates + ");\n"; in GetReduceKernelCode()