/external/tensorflow/tensorflow/core/util/ |
D | cuda_kernel_helper_test.cu.cc | 157 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/ |
D | depthtospace_op_gpu.cu.cc | 165 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 ()()
|
D | spacetodepth_op_gpu.cu.cc | 161 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 ()()
|
D | inplace_ops_functor_gpu.cu.cc | 52 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()
|
D | scan_ops_gpu.h | 246 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()
|
D | bias_op_gpu.cu.cc | 88 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()
|
D | segment_reduction_ops_gpu.cu.cc | 141 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 ()()
|
D | cwise_op_clip_gpu.cu.cc | 65 TF_CHECK_OK(CudaLaunchKernel( in operator ()() 81 TF_CHECK_OK(CudaLaunchKernel( in operator ()() 97 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
|
D | resize_bilinear_op_gpu.cu.cc | 284 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 ()()
|
D | resize_nearest_neighbor_op_gpu.cu.cc | 178 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 ()()
|
D | conv_2d_gpu.h | 440 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>,
|
D | matrix_set_diag_op_gpu.cu.cc | 76 TF_CHECK_OK(CudaLaunchKernel(MatrixSetDiagKernel<Scalar>, in Compute() 83 TF_CHECK_OK(CudaLaunchKernel(MatrixCopyInputAndSetDiagKernel<Scalar>, in Compute()
|
D | gather_functor_gpu.cu.h | 95 TF_CHECK_OK(CudaLaunchKernel( 100 TF_CHECK_OK(CudaLaunchKernel(
|
D | crop_and_resize_op_gpu.cu.cc | 374 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 ()()
|
D | searchsorted_op_gpu.cu.cc | 71 TF_CHECK_OK(CudaLaunchKernel( in Compute() 91 TF_CHECK_OK(CudaLaunchKernel( in Compute()
|
D | reduction_gpu_kernels.cu.h | 514 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 …]
|
D | determinant_op_gpu.cu.cc | 133 TF_CHECK_OK(CudaLaunchKernel( in operator ()() 155 TF_CHECK_OK(CudaLaunchKernel( in operator ()()
|
D | dilation_ops_gpu.cu.cc | 233 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 ()()
|
D | scatter_functor_gpu.cu.h | 130 TF_CHECK_OK(CudaLaunchKernel( 151 TF_CHECK_OK(CudaLaunchKernel(
|
D | adjust_saturation_op_gpu.cu.cc | 39 TF_CHECK_OK(CudaLaunchKernel(internal::adjust_hsv_nhwc<false, true, false, T>, in operator ()()
|
D | adjust_hue_op_gpu.cu.cc | 38 TF_CHECK_OK(CudaLaunchKernel(internal::adjust_hsv_nhwc<true, false, false, T>, in operator ()()
|
D | check_numerics_op_gpu.cu.cc | 69 TF_CHECK_OK(CudaLaunchKernel(CheckNumericsKernel<T>, num_blocks, block_size, in Run()
|
/external/tensorflow/tensorflow/contrib/resampler/kernels/ |
D | resampler_ops_gpu.cu.cc | 122 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/ |
D | cuda_op_kernel.cu.cc | 29 TF_CHECK_OK(::tensorflow::CudaLaunchKernel(AddOneKernel, 32, 256, 0, nullptr, in AddOneKernelLauncher()
|
/external/tensorflow/tensorflow/tools/ci_build/builds/user_ops/ |
D | cuda_op_kernel.cu.cc | 29 TF_CHECK_OK(::tensorflow::CudaLaunchKernel(AddOneKernel, 32, 256, 0, nullptr, in AddOneKernelLauncher()
|