Lines Matching defs:captures
135 SmallVector<Value> captures;
145 ArrayRef<BlockArgument> captures();
161 llvm::SmallVector<Value> captures;
181 ArrayRef<BlockArgument> ParallelComputeFunctionArgs::captures() {
208 // Returns a function type and implicit captures for a parallel compute
209 // function. We'll need a list of implicit captures to setup block and value
214 llvm::SetVector<Value> captures;
215 getUsedValuesDefinedAbove(op.getRegion(), op.getRegion(), captures);
218 inputs.reserve(2 + 4 * op.getNumLoops() + captures.size());
239 // Types of the implicit captures.
240 for (Value capture : captures)
243 // Convert captures to vector for later convenience.
244 SmallVector<Value> capturesVector(captures.begin(), captures.end());
306 // Remaining arguments are implicit captures of the parallel operation.
307 ArrayRef<BlockArgument> captures = args.captures();
425 mapping.map(computeFuncType.captures, captures);
437 return {op.getNumLoops(), func, std::move(computeFuncType.captures)};
598 operands.append(parallelComputeFunction.captures);
679 computeFuncOperands.append(parallelComputeFunction.captures);