Searched refs:axis_to_coord (Results 1 – 2 of 2) sorted by relevance
41 std::map<Axis, std::string> axis_to_coord = { in GetConcatKernelCode() local54 src_coords.push_back(axis_to_coord[axis]); in GetConcatKernelCode()58 dst_coords.push_back(axis_to_coord[axis]); in GetConcatKernelCode()92 c += " int coord = " + axis_to_coord[attr.axis] + ";\n"; in GetConcatKernelCode()
257 std::map<Axis, std::string> axis_to_coord = { in GetReduceKernelCode() local266 dst_check += "DST_" + axis_to_coord[axis] + " >= args.dst_tensor." + in GetReduceKernelCode()275 src_coords[a] = "DST_" + axis_to_coord[a]; in GetReduceKernelCode()310 const std::string src_coord = "SRC_" + axis_to_coord[axis]; in GetReduceKernelCode()414 dst_coordinates += "DST_" + axis_to_coord[a]; in GetReduceKernelCode()