| 426 | } // namespace |
| 427 | |
| 428 | bool ReduceKernel::IsAvailable(TContext* context) const { |
| 429 | auto data_type = context->getAttrStr("data_type"); |
| 430 | auto operand_type = context->getAttrOprand("operand:0").dtype; |
| 431 | if (data_type == "DEFAULT" && operand_type == "f32") { |
| 432 | return true; |
| 433 | } |
| 434 | // TODO:implement int quantized operand_type |
| 435 | return false; |
| 436 | } |
| 437 | //! kernel gen |
| 438 | std::string ReduceKernel::GetKernelSymbol(TContext* context) const { |
| 439 | std::stringstream ss; |
nothing calls this directly
no test coverage detected