Searched refs:CustomOperator (Results 1 – 2 of 2) sorted by relevance
47 class CustomOperator : public Operator {49 CustomOperator(const string &name, const string &type) : Operator(name, type) {} in CustomOperator() function51 ~CustomOperator() override{}; in ~CustomOperator()65 using CusOperatorPtr = std::shared_ptr<ge::CustomOperator>;66 using CustomOperator = ge::CustomOperator; variable
94 auto op = std::make_shared<ge::CustomOperator>(node->fullname_with_scope(), prim->name()); in GenerateCustomOp()153 auto cus_op = std::dynamic_pointer_cast<CustomOperator>(op); in setInput()201 auto cus_op = std::dynamic_pointer_cast<CustomOperator>(op); in setInput()294 auto cus_op = std::dynamic_pointer_cast<CustomOperator>(op); in UpdateSingleOutputDesc()341 output_size = GetCustomOpOutputSize(std::dynamic_pointer_cast<CustomOperator>(op)); in UpdateMultiOutputDesc()368 …(void)std::dynamic_pointer_cast<CustomOperator>(op)->UpdateOutputDesc((*cus_output_map_)[op->GetOp… in UpdateMultiOutputDesc()456 auto cus_op = std::dynamic_pointer_cast<CustomOperator>(op); in updateInputDesc()590 auto cus_op = std::dynamic_pointer_cast<CustomOperator>(op); in setAttr()