Searched refs:ExecuteOp (Results 1 – 9 of 9) sorted by relevance
124 void operator()(async::ExecuteOp executeOp) { in operator ()()143 auto executeOp = waitOp->getParentOfType<async::ExecuteOp>(); in ~DeferWaitCallback()161 static ValueRange addAsyncTokenResults(async::ExecuteOp &executeOp, in addAsyncTokenResults()174 auto newOp = builder.create<async::ExecuteOp>( in addAsyncTokenResults()198 return isa<async::ExecuteOp, async::AwaitOp>(user); in areAllUsersExecuteOrAwait()220 .Case<async::ExecuteOp>([&](auto executeOp) { in addAsyncDependencyAfter()250 auto executeOp = dyn_cast<async::ExecuteOp>(it->getParentOp()); in addAsyncDependencyAfter()
105 auto executeOp = op->getParentOfType<ExecuteOp>(); in verify()123 void ExecuteOp::getNumRegionInvocations( in getNumRegionInvocations()130 void ExecuteOp::getSuccessorRegions(Optional<unsigned> index, in getSuccessorRegions()144 void ExecuteOp::build(OpBuilder &builder, OperationState &result, in build()189 static void print(OpAsmPrinter &p, ExecuteOp op) { in print()302 static LogicalResult verify(ExecuteOp op) { in verify()
245 llvm::SmallSet<ExecuteOp, 4> executeOperations; in addAutomaticRefCounting()248 if (auto execute = dyn_cast<ExecuteOp>(op)) in addAutomaticRefCounting()266 for (ExecuteOp execute : executeOperations) { in addAutomaticRefCounting()
84 bool isAsyncUser = isa<ExecuteOp>(user); in optimizeReferenceCounting()89 isAsyncUser |= isa<ExecuteOp>(user); in optimizeReferenceCounting()
248 auto execute = b.create<ExecuteOp>( in matchAndRewrite()
211 Status ExecuteOp(CallOptions* call_opts, const Operation& operation,
510 Status EagerServiceImpl::ExecuteOp(CallOptions* call_opts, in ExecuteOp() function in tensorflow::eager::EagerServiceImpl571 s = ExecuteOp(call_opts, item.operation(), context->Context(), &executor, in Enqueue()
468 outlineExecuteOp(SymbolTable &symbolTable, ExecuteOp execute) { in outlineExecuteOp()826 WalkResult outlineResult = module.walk([&](ExecuteOp execute) { in runOnOperation()
101 Async_Op<"yield", [HasParent<"ExecuteOp">, NoSideEffect, Terminator]> {