Searched defs:execute_op (Results 1 – 7 of 7) sorted by relevance
47 module.walk([&](tfrt::corert::ExecuteOp execute_op) { in runOnOperation()
79 } else if (auto execute_op = llvm::dyn_cast<tfrt::corert::ExecuteOp>(op)) { in GetDevice() local
1771 if (auto execute_op = in GetNumArgs() local
437 OpBuilder* builder, TF::TPUExecuteOp* execute_op) { in BuildExecuteOp()531 Operation* execute_op, OpBuilder* builder) { in AssignDevicesToReplicatedExecute()678 tf_device::ParallelExecuteOp execute_op; in Rewrite() local696 TF::TPUExecuteOp execute_op; in Rewrite() local
380 const ChainedExecuteFn& execute_op, in ExecuteChained()
430 auto execute_op = [&](const xrt::XRTChainedExecuteOp& op, in DoWork() local
560 auto execute_op = [&](const xrt::XRTChainedExecuteOp& op, in DoWork() local