Searched defs:execute_op (Results 1 – 8 of 8) sorted by relevance
50 module.walk([&](tfrt::corert::ExecuteOp execute_op) { in runOnOperation()
80 } else if (auto execute_op = llvm::dyn_cast<tfrt::corert::ExecuteOp>(op)) { in GetDevice() local
1840 if (auto execute_op = in GetNumArgs() local
52 mlir::BlockArgument arg, mlir::TF::TPUExecuteOp execute_op) { in AddPlaceholderDeviceAttributeToResource()
440 OpBuilder* builder, TF::TPUExecuteOp* execute_op) { in BuildExecuteOp()589 Operation* execute_op, OpBuilder* builder) { in AssignDevicesToReplicatedExecute()
382 const ChainedExecuteFn& execute_op, in ExecuteChained()
430 auto execute_op = [&](const xrt::XRTChainedExecuteOp& op, in DoWork() local
562 auto execute_op = [&](const xrt::XRTChainedExecuteOp& op, in DoWork() local