Searched defs:tpu_functions (Results 1 – 1 of 1) sorted by relevance
76 llvm::SmallVectorImpl<mlir::TF::StatefulPartitionedCallOp>* tpu_functions) { in IdentifyTPUFunctions()140 llvm::SmallVector<mlir::TF::StatefulPartitionedCallOp, 4> tpu_functions; in runOnOperation() local