Searched refs:result_buffers (Results 1 – 3 of 3) sorted by relevance
67 llvm::SmallDenseSet<Value> result_buffers; in runOnFunction() local69 result_buffers.insert(func_arg); in runOnFunction()75 result_buffers.insert(operand); in runOnFunction()81 llvm::SmallVector<Value, 4> worklist(result_buffers.begin(), in runOnFunction()82 result_buffers.end()); in runOnFunction()92 if (result_buffers.insert(alias).second) { in runOnFunction()100 if (result_buffers.insert(alias).second) { in runOnFunction()108 if (result_buffers.insert(alias).second) { in runOnFunction()116 if (result_buffers.insert(alias).second) { in runOnFunction()141 if (result_buffers.insert(operation.getOperand(idx)).second) { in runOnFunction()[all …]
129 TF_ASSIGN_OR_RETURN(ScopedShapedBuffer result_buffers, in ExecuteAsyncOnStream()134 run_options->stream(), result_literal, result_buffers)); in ExecuteAsyncOnStream()135 ExecutionOutput result(std::move(result_buffers)); in ExecuteAsyncOnStream()
376 std::vector<ScopedShapedBuffer> result_buffers; in ExecuteParallelAndRegisterResult() local422 result_buffers.push_back(std::move(result)); in ExecuteParallelAndRegisterResult()426 std::move(result_buffers), result_tags[i])); in ExecuteParallelAndRegisterResult()