Home
last modified time | relevance | path

Searched defs:nthreads (Results 1 – 25 of 62) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Dbincount_op_gpu.cu.cc105 __global__ void BincountReduceKernel(const Tidx* in, T* out, const int nthreads, in BincountReduceKernel()
122 const int nthreads = arr.dimension(0); in Compute() local
137 const int nthreads = num_rows * num_cols; in BincountColReduceKernel() local
167 const int nthreads = num_rows * num_cols; in BincountColReduceSharedKernel() local
Dbias_op_gpu.cu.cc54 __global__ void BiasNHWCKernel(int32 nthreads, const T* __restrict__ input, in BiasNHWCKernel()
64 __global__ void BiasNCHWKernel(int32 nthreads, const T* __restrict__ input, in BiasNCHWKernel()
103 __global__ void BiasGradNHWC_Naive(int32 nthreads, in BiasGradNHWC_Naive()
115 __global__ void BiasGradNCHW_Naive(int32 nthreads, in BiasGradNCHW_Naive()
128 int32 nthreads, const T* __restrict__ output_backprop, in BiasGradNHWC_SharedAtomics()
Dmaxpooling_op_gpu.cu.cc68 const int nthreads, const dtype* __restrict__ bottom_data, in MaxPoolForwardNCHW()
112 const int nthreads, const int32* __restrict__ bottom_data, const int height, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
145 const int nthreads, const dtype* __restrict__ bottom_data, const int height, in MaxPoolForwardNHWC()
203 __global__ void MaxPoolBackward(const int nthreads, in MaxPoolBackward()
231 const int nthreads, const dtype* __restrict__ bottom_data, in MaxPoolGradBackwardNoMaskNCHW()
272 const int nthreads, const dtype* __restrict__ bottom_data, in MaxPoolGradBackwardNoMaskNHWC()
332 __global__ void MaxPoolGradBackward(const int nthreads, in MaxPoolGradBackward()
Ddepthtospace_op_gpu.cu.cc33 __global__ void D2S_NHWC(const int32 nthreads, in D2S_NHWC()
64 __global__ void D2S_NCHW(const int32 nthreads, in D2S_NCHW()
101 __global__ void D2S_NCHW_LOOP(const int32 nthreads, in D2S_NCHW_LOOP()
Dspacetodepth_op_gpu.cu.cc32 __global__ void S2D_NHWC(const int32 nthreads, in S2D_NHWC()
64 __global__ void S2D_NCHW(const int32 nthreads, in S2D_NCHW()
102 __global__ void S2D_NCHW_LOOP(const int32 nthreads, in S2D_NCHW_LOOP()
Ddilation_ops_gpu.cu.cc39 const int32 nthreads, const T* __restrict__ input_ptr, in DilationKernel()
78 const int32 nthreads, const T* __restrict__ input_ptr, in DilationBackpropInputKernel()
128 const int32 nthreads, const T* __restrict__ input_ptr, in DilationBackpropFilterKernel()
Dinplace_ops_functor_gpu.cu.cc30 __global__ void DoParallelConcatOpKernel(int nthreads, const int64 rows, in DoParallelConcatOpKernel()
83 __global__ void DoInplaceOpKernel(int nthreads, const int64 rows, in DoInplaceOpKernel()
/external/grpc-grpc/test/core/gpr/
Dcpu_test.cc59 int nthreads; member
115 uint32_t nthreads = ct.ncores * 3; in cpu_test() local
/external/tensorflow/tensorflow/core/kernels/image/
Dresize_nearest_neighbor_op_gpu.cu.cc36 const int nthreads, const T* __restrict__ bottom_data, const int in_height, in ResizeNearestNeighborNHWC()
67 const int nthreads, const T* __restrict__ bottom_data, const int in_height, in LegacyResizeNearestNeighborNHWC()
96 const int nthreads, const T* __restrict__ top_diff, const int in_height, in ResizeNearestNeighborBackwardNHWC()
127 const int nthreads, const T* __restrict__ top_diff, const int in_height, in LegacyResizeNearestNeighborBackwardNHWC()
Dresize_bilinear_op_gpu.cu.cc118 const int32 nthreads, const T* __restrict__ images, float height_scale, in ResizeBilinearKernel()
168 __global__ void ResizeBilinearGradKernel(const int32 nthreads, in ResizeBilinearGradKernel()
234 const int32 nthreads, const float* __restrict__ input_grad, in ResizeBilinearDeterministicGradKernel()
287 const int32 nthreads, const T* __restrict__ images, float height_scale, in LegacyResizeBilinearKernel()
337 const int32 nthreads, const float* __restrict__ input_grad, in LegacyResizeBilinearGradKernel()
/external/llvm-project/openmp/runtime/test/api/
Domp_get_num_threads.c10 int nthreads = 0; in test_omp_get_num_threads() local
Dkmp_set_defaults_lock_bug.c15 int nthreads = 0; in test_kmp_set_defaults_lock_bug() local
Domp_pause_resource.c6 int fails, nthreads, my_dev; in test_omp_pause_resource() local
/external/llvm-project/openmp/runtime/test/master/
Domp_master.c7 int nthreads; in test_omp_master() local
Domp_master_3.c7 int nthreads; in test_omp_master_3() local
/external/llvm-project/openmp/runtime/test/parallel/
Domp_parallel_num_threads.c9 int nthreads; in test_omp_parallel_num_threads() local
/external/python/cpython2/Tools/ccbench/
Dccbench.py196 def run_throughput_test(func, args, nthreads): argument
300 def run_latency_test(func, args, nthreads): argument
432 def run_bandwidth_test(func, args, nthreads): argument
/external/python/cpython3/Tools/ccbench/
Dccbench.py187 def run_throughput_test(func, args, nthreads): argument
294 def run_latency_test(func, args, nthreads): argument
427 def run_bandwidth_test(func, args, nthreads): argument
/external/ltp/testcases/realtime/perf/latency/
Dpthread_cond_many.c55 int nthreads = 0; variable
186 void test_signal(long iter, long nthreads) in test_signal()
/external/llvm-project/openmp/runtime/test/teams/
Dteams.c29 int nthreads = omp_get_num_threads(); in main() local
/external/llvm-project/lldb/test/API/commands/process/attach-resume/
Dmain.cpp26 static const size_t nthreads = 16; in main() local
/external/compiler-rt/test/asan/TestCases/Posix/
Dhalt_on_error-torture.cc29 size_t nthreads = 10; variable
/external/llvm-project/compiler-rt/test/asan/TestCases/Posix/
Dhalt_on_error-torture.cpp24 size_t nthreads = 10; variable
/external/ltp/testcases/open_posix_testsuite/stress/threads/pthread_create/
Ds-c1.c104 int nthreads; member
147 int nthreads, ctl, i, tmp; in main() local
/external/ltp/testcases/kernel/fs/fs_fill/
Dfs_fill.c23 static unsigned int nthreads; variable

123