| 247 | } // namespace |
| 248 | |
| 249 | bool CvtColorKernel::IsCVAvailable(TContext* context) const { |
| 250 | auto src_dtype = context->getAttrOprand("operand:0").dtype; |
| 251 | bool dtype_ok = Utils::is_int_dtype(src_dtype, 8) || |
| 252 | (context->getAttrStr("mode") == "GRAY2RGB" && src_dtype == "f32"); |
| 253 | bool mode_ok = g_cvt_reg.usable(context->getAttrStr("mode")); |
| 254 | return dtype_ok && mode_ok; |
| 255 | } |
| 256 | |
| 257 | //! kernel gen |
| 258 | std::string CvtColorKernel::GetCVKernelSubSymbol(TContext* context) const { |
nothing calls this directly
no test coverage detected