| /external/dagger2/java/dagger/internal/codegen/writing/ |
| D | ProducerFactoryGenerator.java | 189 ImmutableList<DependencyRequest> asyncDependencies = asyncDependencies(binding); in write() local
190 for (DependencyRequest dependency : asyncDependencies) { in write()
201 FutureTransform futureTransform = FutureTransform.create(fields, binding, asyncDependencies); in write()
288 private static ImmutableList<DependencyRequest> asyncDependencies(Binding binding) { in asyncDependencies() method in ProducerFactoryGenerator
348 ImmutableList<DependencyRequest> asyncDependencies) { in create() argument
349 if (asyncDependencies.isEmpty()) { in create()
351 } else if (asyncDependencies.size() == 1) { in create()
353 fields, binding, Iterables.getOnlyElement(asyncDependencies)); in create()
355 return new MultiArgFutureTransform(fields, binding, asyncDependencies); in create()
436 private final ImmutableList<DependencyRequest> asyncDependencies; field in ProducerFactoryGenerator.MultiArgFutureTransform
[all …]
|
| /external/dagger2/java/dagger/producers/internal/ |
| D | AbstractProducesMethodProducer.java | 75 public final ListenableFuture<T> apply(D asyncDependencies) throws Exception { in apply() argument
81 return callProducesMethod(asyncDependencies); in apply()
91 protected abstract ListenableFuture<T> callProducesMethod(D asyncDependencies) throws Exception; in callProducesMethod() argument
|
| /external/llvm-project/mlir/lib/Conversion/GPUCommon/ |
| D | ConvertLaunchFuncToRuntimeCalls.cpp | 369 auto stream = adaptor.asyncDependencies().front(); in matchAndRewrite()
401 Value stream = adaptor.asyncDependencies().front(); in matchAndRewrite()
446 for (auto pair : llvm::zip(waitOp.asyncDependencies(), operands)) { in matchAndRewrite()
571 if (launchOp.asyncDependencies().size() > 1) in matchAndRewrite()
578 if (!launchOp.asyncToken() && !launchOp.asyncDependencies().empty()) in matchAndRewrite()
616 adaptor.asyncDependencies().empty() in matchAndRewrite()
618 : adaptor.asyncDependencies().front(); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/GPU/IR/ |
| D | GPUDialect.cpp | 449 return getNumOperands() - asyncDependencies().size() - kNumConfigOperands; in getNumKernelOperands()
459 return getOperand(asyncDependencies().size() + kNumConfigOperands + i); in getKernelOperand()
463 auto operands = getOperands().drop_front(asyncDependencies().size()); in getGridSizeOperandValues()
468 auto operands = getOperands().drop_front(asyncDependencies().size()); in getBlockSizeOperandValues()
846 SmallVectorImpl<OpAsmParser::OperandType> &asyncDependencies) { in parseAsyncDependencies() argument
853 return parser.parseOperandList(asyncDependencies, in parseAsyncDependencies()
859 OperandRange asyncDependencies) { in printAsyncDependencies() argument
862 if (asyncDependencies.empty()) in printAsyncDependencies()
865 llvm::interleaveComma(asyncDependencies, printer); in printAsyncDependencies()
|
| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | AsyncRegionRewriter.cpp | 144 auto numDependencies = waitOp.asyncDependencies().size(); in ~DeferWaitCallback()
149 waitOp.asyncDependencies()); in ~DeferWaitCallback()
|
| /external/llvm-project/mlir/include/mlir/Dialect/GPU/ |
| D | GPUOps.td | 296 Arguments<(ins Variadic<GPU_AsyncToken>:$asyncDependencies,
418 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies)
799 let arguments = (ins Variadic<GPU_AsyncToken>:$asyncDependencies);
803 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies) attr-dict
832 let arguments = (ins Variadic<GPU_AsyncToken>:$asyncDependencies,
842 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies) ` `
872 let arguments = (ins Variadic<GPU_AsyncToken>:$asyncDependencies,
877 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies)
|
| D | GPUBase.td | 83 return op.asyncDependencies();
|
| /external/dagger2/javatests/dagger/producers/internal/ |
| D | AbstractProducesMethodProducerTest.java | 144 protected ListenableFuture<T> callProducesMethod(Void asyncDependencies) { in callProducesMethod() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_kernel_to_llvm_pass.cc | 147 if (!launch_op.asyncDependencies().empty() || launch_op.asyncToken()) { in matchAndRewrite()
|