- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for MarkCompilationOps (0.11 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/move_tpu_compile_to_front.cc
struct MoveTpuCompileToFrontPass : public impl::MoveTpuCompileToFrontPassBase<MoveTpuCompileToFrontPass> { void runOnOperation() override; }; void MarkCompilationOps(Operation* func) { func->walk([&](Operation* op) { if (llvm::isa<TF::_TPUCompileMlirOp>(op)) { op->setAttr("_is_compilation", UnitAttr::get(func->getContext())); op = op->getParentOp();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 17 00:26:18 UTC 2023 - 3.2K bytes - Viewed (0)