Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
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()
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()
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()
185 const int nthreads, const dtype* __restrict__ bottom_data, const int height, in MaxPoolBackwardNoMaskNHWC()
245 __global__ void MaxPoolBackward(const int nthreads, in MaxPoolBackward()
274 const int nthreads, const dtype* __restrict__ bottom_data, in MaxPoolGradBackwardNoMaskNCHW()
315 const int nthreads, const dtype* __restrict__ bottom_data, in MaxPoolGradBackwardNoMaskNHWC()
376 __global__ void MaxPoolGradBackward(const int nthreads, in MaxPoolGradBackward()
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()
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()
Dresize_bilinear_op_gpu.cu.cc117 const int32 nthreads, const T* __restrict__ images, float height_scale, in ResizeBilinearKernel()
167 __global__ void ResizeBilinearGradKernel(const int32 nthreads, in ResizeBilinearGradKernel()
233 const int32 nthreads, const T* __restrict__ images, float height_scale, in LegacyResizeBilinearKernel()
283 const int32 nthreads, const float* __restrict__ input_grad, in LegacyResizeBilinearGradKernel()
Dtile_functor_gpu.h33 __global__ void TileKernel(int nthreads, const T* __restrict__ src, in TileKernel()
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()
Droll_op_gpu.cu.cc33 __global__ void RollKernel(const int32 nthreads, const int32 num_dims, in RollKernel()
Dmultinomial_op_gpu.cu.cc47 __global__ void MultinomialKernel(int32 nthreads, const int32 num_classes, in MultinomialKernel()
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()
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()
Dspacetobatch_functor_gpu.cu.cc47 __global__ void S2B(const int32 nthreads, T* __restrict__ space_tensor_ptr, in S2B()
/external/grpc-grpc/test/core/gpr/
Dcpu_test.cc59 int nthreads; member
115 uint32_t nthreads = ct.ncores * 3; in cpu_test() local
/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/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.py194 def run_throughput_test(func, args, nthreads): argument
301 def run_latency_test(func, args, nthreads): argument
434 def run_bandwidth_test(func, args, nthreads): argument
/external/compiler-rt/test/asan/TestCases/Posix/
Dhalt_on_error-torture.cc29 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
/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 long do_threads_test(int nthreads, mes_t * measure) in do_threads_test()
/external/python/cpython2/Python/
Dthread_sgi.h18 static int nthreads; /* protected by count_lock */ variable
/external/eigen/test/
Dcuda_basic.cu146 int nthreads = 100; in test_cuda_basic() local

12