Home
last modified time | relevance | path

Searched refs:composite_func_ops (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_ophint_func_op.cc76 llvm::StringMap<FuncOp> composite_func_ops; in FindCompositeFuncOps() local
79 composite_func_ops[func.getName()] = func; in FindCompositeFuncOps()
81 return composite_func_ops; in FindCompositeFuncOps()
226 LogicalResult ConvertCallOps(llvm::StringMap<FuncOp>* composite_func_ops, in ConvertCallOps() argument
236 auto it = composite_func_ops->find(call_op.getCallee()); in ConvertCallOps()
237 if (it == composite_func_ops->end()) return failure(); in ConvertCallOps()
250 composite_func_ops->erase(it); in ConvertCallOps()
274 llvm::StringMap<FuncOp> composite_func_ops = FindCompositeFuncOps(module); in runOnModule() local
275 if (composite_func_ops.empty()) return; in runOnModule()
276 if (failed(ConvertCallOps(&composite_func_ops, &module))) { in runOnModule()