Searched refs:AddFloatAttr (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | nms_utils.cc | 113 failed(AddFloatAttr(func, attrs, "nms_score_threshold", &fbb)) || in CreateNMSCustomOptions()
114 failed(AddFloatAttr(func, attrs, "nms_iou_threshold", &fbb)) || in CreateNMSCustomOptions()
115 failed(AddFloatAttr(func, attrs, "y_scale", &fbb)) || in CreateNMSCustomOptions()
116 failed(AddFloatAttr(func, attrs, "x_scale", &fbb)) || in CreateNMSCustomOptions()
117 failed(AddFloatAttr(func, attrs, "h_scale", &fbb)) || in CreateNMSCustomOptions()
118 failed(AddFloatAttr(func, attrs, "w_scale", &fbb))) in CreateNMSCustomOptions()
146 LogicalResult ConvertSSDPostProcessFunc::AddFloatAttr( in AddFloatAttr() function in mlir::TFL::ConvertSSDPostProcessFunc
|
| D | nms_utils.h | 65 LogicalResult AddFloatAttr(FuncOp func, DictionaryAttr attrs,
|