Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/kernels/
Dbincount_op_gpu.cu.cc116 __global__ void BincountReduceKernel(const Tidx* in, T* out, const int nthreads, in BincountReduceKernel()
133 const int nthreads = arr.dimension(0); in Compute() local
148 const int nthreads = num_rows * num_cols; in BincountColReduceKernel() local
178 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()
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()
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/pytorch/aten/src/ATen/
DParallelOpenMP.cpp30 auto nthreads = num_threads.load(); in init_num_threads() local
46 void set_num_threads(int nthreads) { in set_num_threads()
DParallelThreadPoolNative.cpp50 void set_num_interop_threads(int nthreads) { in set_num_interop_threads()
61 int nthreads = num_interop_threads.load(); in get_num_interop_threads() local
DParallelNative.cpp66 int _num_pool_threads(int nthreads) { in _num_pool_threads()
211 void set_num_threads(int nthreads) { in set_num_threads()
243 int nthreads = num_intraop_threads.load(); in get_num_threads() local
DParallelCommon.cpp36 int nthreads = std::stoi(value); in get_env_num_threads() local
/external/sandboxed-api/contrib/c-blosc/utils/
Dutils_blosc.cc35 std::string& compressor, int nthreads) { in Compress()
75 std::ofstream& out_stream, int nthreads) { in Decompress()
/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()
/external/grpc-grpc/test/core/gpr/
Dcpu_test.cc62 int nthreads; member
118 uint32_t nthreads = ct.ncores * 3; in cpu_test() local
/external/ComputeLibrary/src/cpu/kernels/
DCpuWinogradConv2dKernel.cpp31 …el(arm_conv::winograd::WinogradImpl &w_impl, arm_conv::ConvolutionArgs &_c_args, uint32_t nthreads) in CpuWinogradConv2dTransformInputKernel()
68 …el(arm_conv::winograd::WinogradImpl &w_impl, arm_conv::ConvolutionArgs &_c_args, uint32_t nthreads) in CpuWinogradConv2dTransformOutputKernel()
/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/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/ComputeLibrary/src/cpu/operators/
DCpuWinogradConv2d.cpp94 uint32_t nthreads = NEScheduler::get().num_threads(); in get_winograd_kernel_implementation() local
170 uint32_t nthreads = NEScheduler::get().num_threads(); in configure() local
324 const uint32_t nthreads = NEScheduler::get().num_threads(); in run() local
/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
Dbarrier.hpp43 void set_nthreads(unsigned int nthreads) { in set_nthreads()
/external/selinux/policycoreutils/setfiles/
Drestore.c76 int process_glob(char *name, struct restore_opts *opts, size_t nthreads, in process_glob()
/external/pytorch/aten/src/ATen/native/cuda/
DAveragePool2d.cu33 __global__ void avg_pool2d_out_cuda_frame(const int nthreads, in avg_pool2d_out_cuda_frame()
82 __global__ void avg_pool2d_out_cuda_frame_nhwc(const int nthreads, in avg_pool2d_out_cuda_frame_nhwc()
131 __global__ void avg_pool2d_backward_out_cuda_frame(const index_t nthreads, const scalar_t* const to… in avg_pool2d_backward_out_cuda_frame()
187 __global__ void avg_pool2d_backward_out_cuda_frame_nhwc(const index_t nthreads, in avg_pool2d_backward_out_cuda_frame_nhwc()
/external/apache-velocity-engine/velocity-engine-core/src/test/java/org/apache/velocity/test/
DRenderVelocityTemplateTest.java127 int nthreads = 4; in testMultiThreadMultipleEvals() local
/external/ltp/testcases/kernel/fs/fs_fill/
Dfs_fill.c24 static unsigned int nthreads; variable

123