Searched defs:custom_op_name (Results 1 – 5 of 5) sorted by relevance
152 std::string custom_op_name = "custom"; in TEST() local174 std::string custom_op_name = "unresolved_custom_op"; in TEST() local203 std::string custom_op_name = "FlexAdd"; in TEST() local
152 TfLiteRegistration CreateUnresolvedCustomOp(const char* custom_op_name) { in CreateUnresolvedCustomOp()
1453 custom_op_name="", argument1542 def __init__(self, dtype=torch.float16, custom_op_name="") -> None: argument
116 std::string custom_op_name = custom_op.custom_code().str(); in IsWeightOnlyOp() local
103 func::FuncOp func, StringRef custom_op_name, in ConvertTflFusableOp()