Home
last modified time | relevance | path

Searched refs:GpuLaunchKernel (Results 1 – 25 of 68) sorted by relevance

123

/external/tensorflow/tensorflow/core/util/
Dgpu_kernel_helper_test.cu.cc163 TF_CHECK_OK(GpuLaunchKernel(SetOutbufZero, cfg.block_count, \ in TEST_F()
168 TF_CHECK_OK(GpuLaunchKernel(Count1D, cfg.block_count, cfg.thread_per_block, \ in TEST_F()
174 TF_CHECK_OK(GpuLaunchKernel(SetOutbufZero, cfg.block_count, \ in TEST_F()
179 TF_CHECK_OK(GpuLaunchKernel(Count1D, cfg.block_count, cfg.thread_per_block, \ in TEST_F()
216 TF_EXPECT_OK(GpuLaunchKernel(SetOutbufZero, cfg1d.block_count, \ in TEST_F()
221 TF_EXPECT_OK(GpuLaunchKernel(Count2D, cfg.block_count, cfg.thread_per_block, \ in TEST_F()
227 TF_EXPECT_OK(GpuLaunchKernel(SetOutbufZero, cfg1d.block_count, \ in TEST_F()
232 TF_EXPECT_OK(GpuLaunchKernel(Count2D, cfg.block_count, cfg.thread_per_block, \ in TEST_F()
258 TF_EXPECT_OK(GpuLaunchKernel(SetOutbufZero, cfg1d.block_count, \ in TEST_F()
263 TF_EXPECT_OK(GpuLaunchKernel(Count3D, cfg.block_count, cfg.thread_per_block, \ in TEST_F()
[all …]
Dgpu_kernel_helper.h100 Status GpuLaunchKernel(void (*function)(Ts...), dim3 grid_dim, dim3 block_dim, in GpuLaunchKernel() function
126 -> decltype(GpuLaunchKernel(std::forward<Args>(args)...)) {
127 return GpuLaunchKernel(std::forward<Args>(args)...);
/external/tensorflow/tensorflow/core/kernels/
Ddepthtospace_op_gpu.cu.cc164 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
200 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
207 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
214 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
229 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
Dmaxpooling_op_gpu.cu.cc407 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
430 GpuLaunchKernel(MaxPoolForwardNHWC<true, T>, in operator ()()
438 GpuLaunchKernel(MaxPoolForwardNHWC<false, T>, in operator ()()
459 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
464 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
481 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
484 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
506 GpuLaunchKernel(MaxPoolGradBackwardNoMaskNHWC<T>, config.block_count, in operator ()()
513 GpuLaunchKernel(MaxPoolGradBackwardNoMaskNCHW<T>, config.block_count, in operator ()()
530 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
Dspacetodepth_op_gpu.cu.cc161 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
197 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
204 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
211 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
226 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
Dinplace_ops_functor_gpu.cu.cc53 TF_CHECK_OK(GpuLaunchKernel( in DoParallelConcatUpdate()
123 TF_CHECK_OK(GpuLaunchKernel(DoInplaceOpKernel<T, I_UPDATE>, in DoInplaceOp()
129 TF_CHECK_OK(GpuLaunchKernel(DoInplaceOpKernel<T, I_ADD>, cfg.block_count, in DoInplaceOp()
135 TF_CHECK_OK(GpuLaunchKernel(DoInplaceOpKernel<T, I_SUB>, cfg.block_count, in DoInplaceOp()
157 TF_CHECK_OK(GpuLaunchKernel(DoInplaceOpKernel<bool, I_UPDATE>, in DoInplaceOp()
Dbias_op_gpu.cu.cc89 TF_CHECK_OK(GpuLaunchKernel(BiasNHWCKernel<T>, config.block_count, in compute()
94 TF_CHECK_OK(GpuLaunchKernel(BiasNCHWKernel<T>, config.block_count, in compute()
233 TF_CHECK_OK(GpuLaunchKernel(BiasGradNHWC_SharedAtomics<T>, in compute()
244 TF_CHECK_OK(GpuLaunchKernel(BiasGradNCHW_SharedAtomics<T>, in compute()
254 TF_CHECK_OK(GpuLaunchKernel( in compute()
258 TF_CHECK_OK(GpuLaunchKernel(BiasGradNCHW_Naive<T>, config.block_count, in compute()
Dresize_nearest_neighbor_op_gpu.cu.cc177 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
184 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
222 TF_CHECK_OK(GpuLaunchKernel(SetZero<T>, output_config.block_count, in operator ()()
232 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
240 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
Dscan_ops_gpu.h272 GpuLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
278 GpuLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
284 GpuLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
290 GpuLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
296 GpuLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
302 GpuLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
Dsegment_reduction_ops_gpu.cu.cc140 TF_CHECK_OK(GpuLaunchKernel(SetZero<T>, config.block_count, in operator ()()
165 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
185 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
202 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
Ddebug_ops_gpu.cu.cc163 TF_CHECK_OK(GpuLaunchKernel(CurtHealthKernel<Tin, Tout>, num_blocks, in Run()
187 TF_CHECK_OK(GpuLaunchKernel(ConciseHealthKernel<Tin, Tout>, num_blocks, in Run()
211 TF_CHECK_OK(GpuLaunchKernel(FullHealthKernel<Tin, Tout>, num_blocks, in Run()
235 TF_CHECK_OK(GpuLaunchKernel(ReduceInfNanThreeSlotsKernel<Tin, Tout>, in Run()
Ddilation_ops_gpu.cu.cc199 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
233 TF_CHECK_OK(GpuLaunchKernel(SetZero<T>, config.block_count, in operator ()()
240 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
275 TF_CHECK_OK(GpuLaunchKernel(SetZero<T>, config.block_count, in operator ()()
282 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
Dcwise_op_clip_gpu.cu.cc73 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
89 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
105 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
Dnon_max_suppression_op.cu.cc288 TF_CHECK_OK(GpuLaunchKernel(SetZero<int>, config.block_count, in NmsGpu()
315 TF_CHECK_OK(GpuLaunchKernel(NMSKernel<true>, block_dim, thread_block, 0, in NmsGpu()
319 TF_CHECK_OK(GpuLaunchKernel(NMSKernel<false>, block_dim, thread_block, 0, in NmsGpu()
334 TF_CHECK_OK(GpuLaunchKernel(Iota<int>, config.block_count, in NmsGpu()
340 TF_CHECK_OK(GpuLaunchKernel(NMSReduce, 1, 1024, bit_mask_len * sizeof(int), in NmsGpu()
477 TF_CHECK_OK(GpuLaunchKernel(Iota<int>, config.block_count, in DoNMS()
498 TF_CHECK_OK(GpuLaunchKernel(IndexMultiSelect<int, float4>, config.block_count, in DoNMS()
540 TF_CHECK_OK(GpuLaunchKernel(SetZero<int>, config.block_count, in DoNMS()
554 TF_CHECK_OK(GpuLaunchKernel( in DoNMS()
Dsplit_lib_gpu.cu.cc206 TF_CHECK_OK(GpuLaunchKernel(SplitOpKernel<T>, config.block_count, in Run()
222 TF_CHECK_OK(GpuLaunchKernel(SplitVOpKernel_fixed<T>, config.block_count, in Run()
235 TF_CHECK_OK(GpuLaunchKernel( in Run()
240 TF_CHECK_OK(GpuLaunchKernel( in Run()
Dcheck_numerics_op_gpu.cu.cc93 TF_CHECK_OK(GpuLaunchKernel(CheckNumericsKernel<T>, num_blocks, block_size, in Run()
111 TF_CHECK_OK(GpuLaunchKernel(CheckNumericsKernelV2<T>, num_blocks, in Run()
Dconv_2d_gpu.h450 TF_CHECK_OK(GpuLaunchKernel(ShuffleInTensor3Simple<T, 2, 1, 0>,
456 TF_CHECK_OK(GpuLaunchKernel(ShuffleInTensor3Simple<T, 1, 2, 0>,
487 TF_CHECK_OK(GpuLaunchKernel(ShuffleInTensor3Simple<T, 2, 1, 0>,
501 TF_CHECK_OK(GpuLaunchKernel(ShuffleInTensor3Simple<T, 2, 0, 1>,
538 TF_CHECK_OK(GpuLaunchKernel(
543 TF_CHECK_OK(GpuLaunchKernel(
655 TF_CHECK_OK(GpuLaunchKernel(
661 TF_CHECK_OK(GpuLaunchKernel(
966 TF_CHECK_OK(GpuLaunchKernel(
978 TF_CHECK_OK(GpuLaunchKernel(ShuffleInTensor3Simple<T, 0, 2, 1, conjugate>,
[all …]
Ddiag_op_gpu.cu.cc65 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
102 GpuLaunchKernel(DiagPartGpuKernel<T>, diag_config.block_count, in operator ()()
Dgather_functor_gpu.cu.h97 TF_CHECK_OK(GpuLaunchKernel(
102 TF_CHECK_OK(GpuLaunchKernel(
Dbucketize_op_gpu.cu.cc100 TF_CHECK_OK(GpuLaunchKernel(BucketizeCustomKernel<T, true>, in Compute()
106 TF_CHECK_OK(GpuLaunchKernel( in Compute()
Dcrop_and_resize_op_gpu.cu.cc376 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
412 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
427 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
462 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
471 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
Dsearchsorted_op_gpu.cu.cc71 TF_CHECK_OK(GpuLaunchKernel( in Compute()
90 TF_CHECK_OK(GpuLaunchKernel( in Compute()
Dconcat_lib_gpu_impl.cu.cc147 TF_CHECK_OK(GpuLaunchKernel( in ConcatGPUImpl()
161 TF_CHECK_OK(GpuLaunchKernel( in ConcatGPUImpl()
167 TF_CHECK_OK(GpuLaunchKernel( in ConcatGPUImpl()
Dstateful_random_ops_gpu.cu.cc93 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
107 TF_CHECK_OK(GpuLaunchKernel(SkipKernel, 1, 1, 0, d.stream(), delta, in operator ()()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dkernels_gpu.cu.cc170 TF_CHECK_OK(GpuLaunchKernel(SparseTensorToCOOMatrixKernel<2>, in operator ()()
175 TF_CHECK_OK(GpuLaunchKernel(SparseTensorToCOOMatrixKernel<3>, in operator ()()
246 TF_CHECK_OK(GpuLaunchKernel(COOMatrixToSparseTensorKernel2D, in operator ()()
263 GpuLaunchKernel(COOMatrixToSparseTensorKernel3D, config.block_count, in operator ()()
332 TF_CHECK_OK(GpuLaunchKernel( in CSRSparseMatrixBatchMulVecImpl()
447 TF_CHECK_OK(GpuLaunchKernel(CSRSparseMatrixSoftmaxKernel2D<T>, in CSRSparseMatrixSoftmaxGPUImpl()
468 TF_CHECK_OK(GpuLaunchKernel(CSRSparseMatrixSoftmaxKernel3D<T>, in CSRSparseMatrixSoftmaxGPUImpl()
642 TF_CHECK_OK(GpuLaunchKernel( in CSRSparseMatrixSoftmaxGradGPUImpl()
673 TF_CHECK_OK(GpuLaunchKernel( in CSRSparseMatrixSoftmaxGradGPUImpl()

123