|
|
|
@ -68,13 +68,18 @@ std::string GpuKernelFactory::SupportedTypeList(const std::string &kernel_name)
|
|
|
|
|
return type_lists;
|
|
|
|
|
}
|
|
|
|
|
for (size_t attr_index = 0; attr_index < (iter->second).size(); ++attr_index) {
|
|
|
|
|
std::string type_list = "[";
|
|
|
|
|
std::string type_list = "in[";
|
|
|
|
|
auto attr = (iter->second)[attr_index].first;
|
|
|
|
|
for (size_t input_index = 0; input_index < attr.GetInputSize(); ++input_index) {
|
|
|
|
|
type_list = type_list + TypeId2String(attr.GetInputAttr(input_index).first) +
|
|
|
|
|
((input_index == (attr.GetInputSize() - 1)) ? "" : " ");
|
|
|
|
|
}
|
|
|
|
|
type_lists = type_lists + type_list + "] ";
|
|
|
|
|
type_list = type_list + "], out[";
|
|
|
|
|
for (size_t input_index = 0; input_index < attr.GetOutputSize(); ++input_index) {
|
|
|
|
|
type_list = type_list + TypeId2String(attr.GetOutputAttr(input_index).first) +
|
|
|
|
|
((input_index == (attr.GetOutputSize() - 1)) ? "" : " ");
|
|
|
|
|
}
|
|
|
|
|
type_lists = type_lists + type_list + "]; ";
|
|
|
|
|
}
|
|
|
|
|
return type_lists;
|
|
|
|
|
}
|
|
|
|
|