namespace
| 603 | |
| 604 | } // namespace |
| 605 | bool ReduceKernel::IsAvailable(TContext* context) const { |
| 606 | auto data_type = context->getAttrStr("data_type"); |
| 607 | auto operand_type = context->getAttrOprand("operand:0").dtype; |
| 608 | if (data_type == "DEFAULT" && (operand_type == "f32" || operand_type == "f16")) { |
| 609 | return true; |
| 610 | } |
| 611 | // TODO:implement int quantized operand_type |
| 612 | return false; |
| 613 | } |
| 614 | //! kernel gen |
| 615 | std::string ReduceKernel::GetKernelSymbol(TContext* context) const { |
| 616 | std::stringstream ss; |
nothing calls this directly
no test coverage detected