Home
last modified time | relevance | path

Searched defs:nthreads (Results 1 – 25 of 55) 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()
106 __global__ void BiasGradNHWC_Naive(int32 nthreads, in BiasGradNHWC_Naive()
118 __global__ void BiasGradNCHW_Naive(int32 nthreads, in BiasGradNCHW_Naive()
131 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()
Dtile_functor_gpu.h33 __global__ void TileKernel(int nthreads, const T* __restrict__ src, in TileKernel()
Droll_op_gpu.cu.cc33 __global__ void RollKernel(const int32 nthreads, const int32 num_dims, in RollKernel()
Dpooling_ops_3d_gpu.cu.cc31 const int nthreads, const dtype* __restrict__ bottom_data, in MaxPoolGradBackwardNoMaskNCDHW()
82 const int nthreads, const dtype* __restrict__ bottom_data, in MaxPoolGradBackwardNoMaskNDHWC()
Dmultinomial_op_gpu.cu.cc42 __global__ void MultinomialKernel(int32 nthreads, const int32 num_classes, in MultinomialKernel()
/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.cc117 const int32 nthreads, const T* __restrict__ images, float height_scale, in ResizeBilinearKernel()
166 __global__ void ResizeBilinearGradKernel(const int32 nthreads, in ResizeBilinearGradKernel()
232 const int32 nthreads, const float* __restrict__ input_grad, in ResizeBilinearDeterministicGradKernel()
285 const int32 nthreads, const T* __restrict__ images, float height_scale, in LegacyResizeBilinearKernel()
335 const int32 nthreads, const float* __restrict__ input_grad, in LegacyResizeBilinearGradKernel()
Dcrop_and_resize_op_gpu.cu.cc41 const int32 nthreads, const T* __restrict__ image_ptr, in CropAndResizeKernel()
134 const int32 nthreads, const float* __restrict__ grads_ptr, in CropAndResizeBackpropImageKernel()
230 const int32 nthreads, const float* __restrict__ grads_ptr, in CropAndResizeBackpropBoxesKernel()
/external/grpc-grpc/test/core/gpr/
Dcpu_test.cc59 int nthreads; member
115 uint32_t nthreads = ct.ncores * 3; in cpu_test() local
/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/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/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/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/compiler-rt/test/asan/TestCases/Posix/
Dhalt_on_error-torture.cc29 size_t nthreads = 10; variable
/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
Dbarrier.hpp43 void set_nthreads(unsigned int nthreads) { in set_nthreads()
/external/selinux/policycoreutils/setfiles/
Drestore.c75 int process_glob(char *name, struct restore_opts *opts, size_t nthreads) in process_glob()
/external/ltp/testcases/kernel/fs/fs_fill/
Dfs_fill.c23 static unsigned int nthreads; variable
/external/autotest/client/tests/monotonic_time/src/
Dthreads.c86 int create_threads(int nthreads, thread_func_t func, void *arg) in create_threads()
/external/ltp/testcases/open_posix_testsuite/stress/threads/pthread_cond_timedwait/
Ds-c.c175 int nthreads; member
290 static long do_threads_test(int nthreads, mes_t * measure) in do_threads_test()

123