Searched refs:tpu_functions (Results 1 – 1 of 1) sorted by relevance
76 llvm::SmallVectorImpl<mlir::TF::StatefulPartitionedCallOp>* tpu_functions) { in IdentifyTPUFunctions() argument92 tpu_functions->emplace_back(call); in IdentifyTPUFunctions()140 llvm::SmallVector<mlir::TF::StatefulPartitionedCallOp, 4> tpu_functions; in runOnOperation() local142 IdentifyTPUFunctions(module, &tpu_meshes, &tpu_functions); in runOnOperation()144 for (auto tpu_function_and_mesh : llvm::zip(tpu_meshes, tpu_functions)) { in runOnOperation()