Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/util/
Dgpu_kernel_helper_test.cu.cc184 TF_CHECK_OK(GpuLaunchKernel(SetOutbufZero, cfg.block_count, \ in TEST_F()
189 TF_CHECK_OK(GpuLaunchKernel(Count1D, cfg.block_count, cfg.thread_per_block, \ in TEST_F()
197 TF_CHECK_OK(GpuLaunchKernel(SetOutbufZero, cfg.block_count, \ in TEST_F()
202 TF_CHECK_OK(GpuLaunchKernel(Count1D, cfg.block_count, cfg.thread_per_block, \ in TEST_F()
241 TF_EXPECT_OK(GpuLaunchKernel(SetOutbufZero, cfg1d.block_count, \ in TEST_F()
246 TF_EXPECT_OK(GpuLaunchKernel(Count2D, cfg.block_count, cfg.thread_per_block, \ in TEST_F()
254 TF_EXPECT_OK(GpuLaunchKernel(SetOutbufZero, cfg1d.block_count, \ in TEST_F()
259 TF_EXPECT_OK(GpuLaunchKernel(Count2D, cfg.block_count, cfg.thread_per_block, \ in TEST_F()
286 TF_EXPECT_OK(GpuLaunchKernel(SetOutbufZero, cfg1d.block_count, \ in TEST_F()
291 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 ()()
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()
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 ()()
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()
Dmaxpooling_op_gpu.cu.cc363 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
386 GpuLaunchKernel(MaxPoolForwardNHWC<true, T>, in operator ()()
394 GpuLaunchKernel(MaxPoolForwardNHWC<false, T>, in operator ()()
412 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
415 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
437 GpuLaunchKernel(MaxPoolGradBackwardNoMaskNHWC<T>, config.block_count, in operator ()()
444 GpuLaunchKernel(MaxPoolGradBackwardNoMaskNCHW<T>, config.block_count, in operator ()()
461 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
Dscan_ops_gpu.h259 GpuLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
265 GpuLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
271 GpuLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
277 GpuLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
289 GpuLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
295 GpuLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
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 ()()
Dsegment_reduction_ops_gpu.cu.cc147 TF_CHECK_OK(GpuLaunchKernel(SetToValue<T>, config.block_count, in operator ()()
172 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
194 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
211 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
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()
Dsplit_lib_gpu.cu.cc202 TF_CHECK_OK(GpuLaunchKernel(SplitOpKernel<T>, config.block_count, in Run()
218 TF_CHECK_OK(GpuLaunchKernel(SplitVOpKernel_fixed<T>, config.block_count, in Run()
231 TF_CHECK_OK(GpuLaunchKernel( in Run()
236 TF_CHECK_OK(GpuLaunchKernel( in Run()
Dconv_2d_gpu.h497 TF_CHECK_OK(GpuLaunchKernel(ShuffleInTensor3Simple<T, 2, 1, 0>,
503 TF_CHECK_OK(GpuLaunchKernel(ShuffleInTensor3Simple<T, 1, 2, 0>,
534 TF_CHECK_OK(GpuLaunchKernel(ShuffleInTensor3Simple<T, 2, 1, 0>,
548 TF_CHECK_OK(GpuLaunchKernel(ShuffleInTensor3Simple<T, 2, 0, 1>,
585 TF_CHECK_OK(GpuLaunchKernel(
591 TF_CHECK_OK(GpuLaunchKernel(
704 TF_CHECK_OK(GpuLaunchKernel(
710 TF_CHECK_OK(GpuLaunchKernel(
1015 TF_CHECK_OK(GpuLaunchKernel(
1027 TF_CHECK_OK(GpuLaunchKernel(ShuffleInTensor3Simple<T, 0, 2, 1, conjugate>,
[all …]
Ddiag_op_gpu.cu.cc65 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
103 GpuLaunchKernel(DiagPartGpuKernel<T>, diag_config.block_count, in operator ()()
Dgather_functor_gpu.cu.h99 TF_CHECK_OK(GpuLaunchKernel(
107 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()
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()
Dbincount_op_gpu.cu.cc126 return GpuLaunchKernel(BincountReduceKernel<Tidx, T>, config.block_count, in Compute()
212 return GpuLaunchKernel( in Compute()
218 return GpuLaunchKernel( in Compute()
Dsearchsorted_op_gpu.cu.cc75 TF_CHECK_OK(GpuLaunchKernel( in Compute()
98 TF_CHECK_OK(GpuLaunchKernel( in Compute()
Drelu_op_gpu.cu.cc183 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
190 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
231 TF_CHECK_OK(GpuLaunchKernel( in operator ()()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dkernels_gpu.cu.cc157 TF_CHECK_OK(GpuLaunchKernel(SparseTensorToCOOMatrixKernel<2>, in operator ()()
162 TF_CHECK_OK(GpuLaunchKernel(SparseTensorToCOOMatrixKernel<3>, in operator ()()
233 TF_CHECK_OK(GpuLaunchKernel(COOMatrixToSparseTensorKernel2D, in operator ()()
250 GpuLaunchKernel(COOMatrixToSparseTensorKernel3D, config.block_count, in operator ()()
319 TF_CHECK_OK(GpuLaunchKernel( in CSRSparseMatrixBatchMulVecImpl()
434 TF_CHECK_OK(GpuLaunchKernel(CSRSparseMatrixSoftmaxKernel2D<T>, in CSRSparseMatrixSoftmaxGPUImpl()
455 TF_CHECK_OK(GpuLaunchKernel(CSRSparseMatrixSoftmaxKernel3D<T>, in CSRSparseMatrixSoftmaxGPUImpl()
629 TF_CHECK_OK(GpuLaunchKernel( in CSRSparseMatrixSoftmaxGradGPUImpl()
660 TF_CHECK_OK(GpuLaunchKernel( in CSRSparseMatrixSoftmaxGradGPUImpl()
/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cu.cc272 TF_CHECK_OK(GpuLaunchKernel(SetZero<int>, config.block_count, in NmsGpu()
299 TF_CHECK_OK(GpuLaunchKernel(NMSKernel<true>, block_dim, thread_block, 0, in NmsGpu()
303 TF_CHECK_OK(GpuLaunchKernel(NMSKernel<false>, block_dim, thread_block, 0, in NmsGpu()
318 TF_CHECK_OK(GpuLaunchKernel(Iota<int>, config.block_count, in NmsGpu()
324 TF_CHECK_OK(GpuLaunchKernel(NMSReduce, 1, 1024, bit_mask_len * sizeof(int), in NmsGpu()
462 TF_CHECK_OK(GpuLaunchKernel(Iota<int>, config.block_count, in DoNMS()
483 TF_CHECK_OK(GpuLaunchKernel(IndexMultiSelect<int, float4>, config.block_count, in DoNMS()
525 TF_CHECK_OK(GpuLaunchKernel(SetZero<int>, config.block_count, in DoNMS()
539 TF_CHECK_OK(GpuLaunchKernel( in DoNMS()
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 ()()

123