| /external/tflite-support/tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/vision/segmenter/ |
| D | ImageSegmenter.java | 187 options.getNumThreads()); in createFromBufferAndOptions()
215 public abstract int getNumThreads();
332 options.getNumThreads());
|
| /external/pytorch/caffe2/utils/threadpool/ |
| D | ThreadPool.cc | 33 int getNumThreads() const override;
156 int ThreadPoolImpl::getNumThreads() const { in getNumThreads() function in caffe2::ThreadPoolImpl
|
| D | ThreadPool.h | 39 virtual int getNumThreads() const = 0;
|
| D | pthreadpool_impl.cc | 72 return reinterpret_cast<caffe2::ThreadPool*>(threadpool)->getNumThreads(); in legacy_pthreadpool_get_threads_count()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Normalization.cuh | 34 static int getNumThreads(int nElem) { in getNumThreads() function
649 int tf = getNumThreads(input.size(2)); in batch_norm_backward_cuda_template()
686 int tf = getNumThreads(input.size(2)); in batch_norm_stats_cuda_template()
723 int tf = std::max<int>(getNumThreads(input.size(2)/4), in batch_norm_elemt_cuda_template()
724 std::min<int>(getNumThreads(input.size(2)), 64)); in batch_norm_elemt_cuda_template()
768 int block = getNumThreads(features); in batch_norm_gather_stats_cuda_template()
826 …int block_x = std::min<int>(std::max<int>(getNumThreads(feature_size), warp_size), MAX_BLOCK_SIZE/… in batch_norm_backward_reduce_cuda_template()
871 int tf = std::max<int>(getNumThreads(input.size(2)/4), in batch_norm_backward_elemt_cuda_template()
872 std::min<int>(getNumThreads(input.size(2)), 64)); in batch_norm_backward_elemt_cuda_template()
922 int tf = std::max<int>(getNumThreads(input.size(2)/4), in batch_norm_backward_elemt_cuda_template()
[all …]
|
| /external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/ |
| D | InterpreterApi.java | 126 public int getNumThreads() { in getNumThreads() method in InterpreterApi.Options
|
| D | NativeInterpreterWrapper.java | 94 modelHandle, errorHandle, options.getNumThreads(), useXnnpack, delegateHandles); in init()
107 modelHandle, errorHandle, options.getNumThreads(), useXnnpack, delegateHandles); in init()
|
| /external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| D | SoftMax.cu | 68 static int getNumThreads(int nElem) { in getNumThreads() function
431 int block_size = getNumThreads(pool_size); in cuda_sparse_coo_softmax()
563 int block_size = getNumThreads(pool_size); in cuda_sparse_coo_softmax_backward()
|
| /external/tflite-support/tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/vision/detector/ |
| D | ObjectDetector.java | 379 public int getNumThreads() {
|
| /external/tflite-support/tensorflow_lite_support/java/src/java/org/tensorflow/lite/task/vision/classifier/ |
| D | ImageClassifier.java | 359 public int getNumThreads() {
|
| /external/pytorch/torch/csrc/distributed/c10d/ |
| D | logger.cpp | 110 auto n_threads = gloo_pg->getNumThreads(); in set_env_variables()
|
| D | ProcessGroupGloo.hpp | 397 int getNumThreads() { in getNumThreads() function in c10d::ProcessGroupGloo
|
| /external/clang/lib/AST/ |
| D | StmtProfile.cpp | 300 if (C->getNumThreads()) in VisitOMPNumThreadsClause()
301 Profiler->VisitStmt(C->getNumThreads()); in VisitOMPNumThreadsClause()
|
| D | StmtPrinter.cpp | 620 Node->getNumThreads()->printPretty(OS, nullptr, Policy, 0); in VisitOMPNumThreadsClause()
|
| /external/clang/include/clang/AST/ |
| D | RecursiveASTVisitor.h | 2625 TRY_TO(TraverseStmt(C->getNumThreads()));
|
| D | OpenMPClause.h | 370 Expr *getNumThreads() const { return cast_or_null<Expr>(NumThreads); } in getNumThreads() function
|
| /external/clang/lib/Serialization/ |
| D | ASTWriterStmt.cpp | 1811 Record.AddStmt(C->getNumThreads()); in VisitOMPNumThreadsClause()
|
| /external/clang/lib/CodeGen/ |
| D | CGStmtOpenMP.cpp | 1200 auto NumThreads = CGF.EmitScalarExpr(NumThreadsClause->getNumThreads(), in emitCommonOMPParallelDirective()
|
| /external/clang/tools/libclang/ |
| D | CIndex.cpp | 2073 Visitor->AddStmt(C->getNumThreads()); in VisitOMPNumThreadsClause()
|
| /external/clang/lib/Sema/ |
| D | TreeTransform.h | 7654 ExprResult NumThreads = getDerived().TransformExpr(C->getNumThreads()); in TransformOMPNumThreadsClause()
|