Searched refs:AddIntAttr (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | nms_utils.cc | 110 if (failed(AddIntAttr(func, attrs, "max_detections", &fbb)) || in CreateNMSCustomOptions()
111 failed(AddIntAttr(func, attrs, "max_classes_per_detection", &fbb)) || in CreateNMSCustomOptions()
112 failed(AddIntAttr(func, attrs, "num_classes", &fbb)) || in CreateNMSCustomOptions()
134 LogicalResult ConvertSSDPostProcessFunc::AddIntAttr( in AddIntAttr() function in mlir::TFL::ConvertSSDPostProcessFunc
|
| D | nms_utils.h | 61 LogicalResult AddIntAttr(FuncOp func, DictionaryAttr attrs,
|