- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for UnwrapTFCustomOps (0.11 sec)
-
tensorflow/compiler/mlir/lite/utils/fake_quant_utils.cc
namespace mlir { namespace TFL { // Moves the TF operations out from the tfl.TFCustomOps wrappers inside the // function. This is a no-op for the ops which are not wrapped. LogicalResult UnwrapTFCustomOps(func::FuncOp fn, OpBuilder& builder) { llvm::SmallVector<Operation*, 4> wrapped_ops; fn.walk([&](TFL::CustomTfOp custom_op) { auto* real_op = &custom_op.getBody().front().front(); wrapped_ops.push_back(real_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 03 00:14:05 UTC 2023 - 4.3K bytes - Viewed (0)