- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for CreateNMSCustomOptions (0.14 sec)
-
tensorflow/compiler/mlir/lite/utils/nms_utils.cc
func_->setAttr(kTFImplements, StringAttr::get(func_.getContext(), kCustomSSDPostprocessing)); OpBuilder builder(func_.getBody()); std::string custom_option_buffer; if (failed(CreateNMSCustomOptions(func_, attr_.getAttrs(), custom_option_buffer))) { return failure(); } auto op = builder.create<CustomOp>( func_.getLoc(), func_.getFunctionType().getResults(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/nms_utils.h
: func_(func), attr_(attr) {} LogicalResult RewriteFunc(); LogicalResult VerifySignature(); private: LogicalResult CreateNMSCustomOptions(func::FuncOp func, DictionaryAttr attrs, std::string& custom_option_buffer); LogicalResult AddIntAttr(func::FuncOp func, DictionaryAttr attrs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 19 00:13:50 UTC 2022 - 2.8K bytes - Viewed (0)