Home
last modified time | relevance | path

Searched refs:CudaLaunchKernel (Results 1 – 25 of 53) sorted by relevance

123

/external/tensorflow/tensorflow/core/util/
Dcuda_kernel_helper_test.cu.cc157 TF_CHECK_OK(CudaLaunchKernel(SetOutbufZero, cfg.block_count, \ in TEST_F()
162 TF_CHECK_OK(CudaLaunchKernel(Count1D, cfg.block_count, cfg.thread_per_block, \ in TEST_F()
168 TF_CHECK_OK(CudaLaunchKernel(SetOutbufZero, cfg.block_count, \ in TEST_F()
173 TF_CHECK_OK(CudaLaunchKernel(Count1D, cfg.block_count, cfg.thread_per_block, \ in TEST_F()
210 TF_EXPECT_OK(CudaLaunchKernel(SetOutbufZero, cfg1d.block_count, \ in TEST_F()
215 TF_EXPECT_OK(CudaLaunchKernel(Count2D, cfg.block_count, \ in TEST_F()
222 TF_EXPECT_OK(CudaLaunchKernel(SetOutbufZero, cfg1d.block_count, \ in TEST_F()
227 TF_EXPECT_OK(CudaLaunchKernel(Count2D, cfg.block_count, \ in TEST_F()
254 TF_EXPECT_OK(CudaLaunchKernel(SetOutbufZero, cfg1d.block_count, \ in TEST_F()
259 TF_EXPECT_OK(CudaLaunchKernel(Count3D, cfg.block_count, \ in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Ddepthtospace_op_gpu.cu.cc165 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
201 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
208 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
215 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
230 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
Dspacetodepth_op_gpu.cu.cc161 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
197 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
204 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
211 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
226 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
Dinplace_ops_functor_gpu.cu.cc52 TF_CHECK_OK(CudaLaunchKernel( in DoParallelConcatUpdate()
120 TF_CHECK_OK(CudaLaunchKernel(DoInplaceOpKernel<T, I_UPDATE>, in DoInplaceOp()
126 TF_CHECK_OK(CudaLaunchKernel(DoInplaceOpKernel<T, I_ADD>, cfg.block_count, in DoInplaceOp()
132 TF_CHECK_OK(CudaLaunchKernel(DoInplaceOpKernel<T, I_SUB>, cfg.block_count, in DoInplaceOp()
154 TF_CHECK_OK(CudaLaunchKernel(DoInplaceOpKernel<bool, I_UPDATE>, in DoInplaceOp()
Dscan_ops_gpu.h246 CudaLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
252 CudaLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
258 CudaLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
264 CudaLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
270 CudaLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
276 CudaLaunchKernel(scan_kernel<T, Op, block_size, items_per_thread>, in LaunchScan()
Dbias_op_gpu.cu.cc88 TF_CHECK_OK(CudaLaunchKernel(BiasNHWCKernel<T>, config.block_count, in compute()
93 TF_CHECK_OK(CudaLaunchKernel(BiasNCHWKernel<T>, config.block_count, in compute()
229 TF_CHECK_OK(CudaLaunchKernel( in compute()
239 TF_CHECK_OK(CudaLaunchKernel( in compute()
243 TF_CHECK_OK(CudaLaunchKernel(BiasGradNCHW_Naive<T>, config.block_count, in compute()
Dsegment_reduction_ops_gpu.cu.cc141 TF_CHECK_OK(CudaLaunchKernel(SetZero<T>, config.block_count, in operator ()()
166 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
187 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
202 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
Dcwise_op_clip_gpu.cu.cc65 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
81 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
97 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
Dresize_bilinear_op_gpu.cu.cc284 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
290 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
322 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
330 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
337 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
Dresize_nearest_neighbor_op_gpu.cu.cc178 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
185 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
232 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
240 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
Dconv_2d_gpu.h440 TF_CHECK_OK(CudaLaunchKernel(ShuffleInTensor3Simple<T, 2, 1, 0>,
462 TF_CHECK_OK(CudaLaunchKernel(ShuffleInTensor3Simple<T, 2, 1, 0>,
493 TF_CHECK_OK(CudaLaunchKernel(
498 TF_CHECK_OK(CudaLaunchKernel(
610 TF_CHECK_OK(CudaLaunchKernel(
616 TF_CHECK_OK(CudaLaunchKernel(
922 TF_CHECK_OK(CudaLaunchKernel(
934 TF_CHECK_OK(CudaLaunchKernel(ShuffleInTensor3Simple<T, 0, 2, 1, conjugate>,
967 TF_CHECK_OK(CudaLaunchKernel(ShuffleInTensor3Simple<T, 2, 1, 0, conjugate>,
Dmatrix_set_diag_op_gpu.cu.cc76 TF_CHECK_OK(CudaLaunchKernel(MatrixSetDiagKernel<Scalar>, in Compute()
83 TF_CHECK_OK(CudaLaunchKernel(MatrixCopyInputAndSetDiagKernel<Scalar>, in Compute()
Dgather_functor_gpu.cu.h95 TF_CHECK_OK(CudaLaunchKernel(
100 TF_CHECK_OK(CudaLaunchKernel(
Dcrop_and_resize_op_gpu.cu.cc374 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
409 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
424 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
459 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
468 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
Dsearchsorted_op_gpu.cu.cc71 TF_CHECK_OK(CudaLaunchKernel( in Compute()
91 TF_CHECK_OK(CudaLaunchKernel( in Compute()
Dreduction_gpu_kernels.cu.h514 TF_CHECK_OK(CudaLaunchKernel(
535 TF_CHECK_OK(CudaLaunchKernel(BlockReduceKernel<IN_T, T*, num_threads, Op>,
544 TF_CHECK_OK(CudaLaunchKernel(CleanupSegments<T*, OUT_T, Op>, 1, 32, 0,
582 TF_CHECK_OK(CudaLaunchKernel(RowReduceKernel<IN_T, OUT_T, Op>, num_blocks,
635 TF_CHECK_OK(CudaLaunchKernel(
645 TF_CHECK_OK(CudaLaunchKernel(ColumnReduceMax16ColumnsKernel<IN_T, T*, Op>,
652 TF_CHECK_OK(CudaLaunchKernel(CleanupSegments<T*, OUT_T, Op>, new_grid_dim,
674 TF_CHECK_OK(CudaLaunchKernel(ColumnReduceKernel<IN_T, OUT_T, Op>, grid_dim,
685 TF_CHECK_OK(CudaLaunchKernel(
691 TF_CHECK_OK(CudaLaunchKernel(CleanupSegments<T*, OUT_T, Op>, new_grid_dim,
[all …]
Ddeterminant_op_gpu.cu.cc133 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
155 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
Ddilation_ops_gpu.cu.cc233 TF_CHECK_OK(CudaLaunchKernel(SetZero<T>, config.block_count, in operator ()()
240 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
275 TF_CHECK_OK(CudaLaunchKernel(SetZero<T>, config.block_count, in operator ()()
Dscatter_functor_gpu.cu.h130 TF_CHECK_OK(CudaLaunchKernel(
151 TF_CHECK_OK(CudaLaunchKernel(
Dadjust_saturation_op_gpu.cu.cc39 TF_CHECK_OK(CudaLaunchKernel(internal::adjust_hsv_nhwc<false, true, false, T>, in operator ()()
Dadjust_hue_op_gpu.cu.cc38 TF_CHECK_OK(CudaLaunchKernel(internal::adjust_hsv_nhwc<true, false, false, T>, in operator ()()
Dcheck_numerics_op_gpu.cu.cc69 TF_CHECK_OK(CudaLaunchKernel(CheckNumericsKernel<T>, num_blocks, block_size, in Run()
/external/tensorflow/tensorflow/contrib/resampler/kernels/
Dresampler_ops_gpu.cu.cc122 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
257 TF_CHECK_OK(::tensorflow::CudaLaunchKernel( in operator ()()
262 TF_CHECK_OK(::tensorflow::CudaLaunchKernel( in operator ()()
269 TF_CHECK_OK(CudaLaunchKernel(ResamplerGrad2DKernel<T>, config.block_count, in operator ()()
/external/tensorflow/tensorflow/examples/adding_an_op/
Dcuda_op_kernel.cu.cc29 TF_CHECK_OK(::tensorflow::CudaLaunchKernel(AddOneKernel, 32, 256, 0, nullptr, in AddOneKernelLauncher()
/external/tensorflow/tensorflow/tools/ci_build/builds/user_ops/
Dcuda_op_kernel.cu.cc29 TF_CHECK_OK(::tensorflow::CudaLaunchKernel(AddOneKernel, 32, 256, 0, nullptr, in AddOneKernelLauncher()

123