Searched refs:dst_coords (Results 1 – 2 of 2) sorted by relevance
53 std::vector<std::string> dst_coords; in GetAveragePoolingKernelCode() local56 dst_coords.push_back(axis_to_dst_coord[axis]); in GetAveragePoolingKernelCode()66 std::string dst_coord = dst_coords[0]; in GetAveragePoolingKernelCode()67 for (int i = 1; i < dst_coords.size(); ++i) { in GetAveragePoolingKernelCode()68 dst_coord += ", " + dst_coords[i]; in GetAveragePoolingKernelCode()177 std::vector<std::string> dst_coords; in GetMaxPoolingKernelCode() local180 dst_coords.push_back(axis_to_dst_coord[axis]); in GetMaxPoolingKernelCode()190 std::string dst_coord = dst_coords[0]; in GetMaxPoolingKernelCode()191 for (int i = 1; i < dst_coords.size(); ++i) { in GetMaxPoolingKernelCode()192 dst_coord += ", " + dst_coords[i]; in GetMaxPoolingKernelCode()
47 std::vector<std::string> dst_coords; in GetConcatKernelCode() local58 dst_coords.push_back(axis_to_coord[axis]); in GetConcatKernelCode()65 std::string dst_coord = dst_coords[0]; in GetConcatKernelCode()66 for (int i = 1; i < dst_coords.size(); ++i) { in GetConcatKernelCode()67 dst_coord += ", " + dst_coords[i]; in GetConcatKernelCode()