Home
last modified time | relevance | path

Searched refs:block_count (Results 1 – 25 of 124) sorted by relevance

12345

/external/tensorflow/tensorflow/core/util/
Dcuda_launch_config.h118 int block_count = -1; member
134 const int block_count = in GetCudaLaunchConfig() local
140 config.block_count = block_count; in GetCudaLaunchConfig()
155 int block_count = 0; in GetCudaLaunchConfig() local
159 &block_count, &thread_per_block, func, dynamic_shared_memory_size, in GetCudaLaunchConfig()
163 block_count = in GetCudaLaunchConfig()
164 std::min(block_count, DivUp(work_element_count, thread_per_block)); in GetCudaLaunchConfig()
168 config.block_count = block_count; in GetCudaLaunchConfig()
182 int block_count = 0; in GetCudaLaunchConfigFixedBlockSize() local
185 &block_count, func, fixed_block_size, dynamic_shared_memory_size); in GetCudaLaunchConfigFixedBlockSize()
[all …]
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()
195 a.block_count.x == b.block_count.x && in operator ==()
196 a.block_count.y == b.block_count.y && in operator ==()
197 a.block_count.z == b.block_count.z && in operator ==()
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()
[all …]
/external/autotest/client/site_tests/platform_DMVerityBitCorruption/
Dplatform_DMVerityBitCorruption.py18 block_count): argument
20 run_count, backing_path, block_size, block_count))
21 run_count = run_count % block_count
30 block_count): argument
32 run_count, backing_path, block_size, block_count))
35 dev.seek(block_count * block_size, os.SEEK_SET)
39 dev.seek(block_count * block_size, os.SEEK_SET)
/external/autotest/client/site_tests/platform_DMVerityCorruption/
Dplatform_DMVerityCorruption.py14 block_count): argument
16 run_count, backing_path, block_size, block_count))
18 run_count = run_count % block_count
22 block_count): argument
24 run_count, backing_path, block_size, block_count))
26 dev.seek(block_count * block_size, os.SEEK_SET)
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dpartition_assignment.h41 LaunchDimensions(int64 block_count, int64 threads_per_block) in LaunchDimensions() argument
42 : block_count_(block_count), threads_per_block_(threads_per_block) {} in LaunchDimensions()
48 int64 block_count() const { return block_count_; } in block_count() function
50 int64 launch_bound() const { return block_count() * threads_per_block(); } in launch_bound()
Dpartition_assignment.cc37 out << absl::StrFormat("[block: %d, thread: %d]", launch_dims.block_count(), in operator <<()
91 int64 block_count = CeilOfRatio(num_elements, threads_per_block); in CalculateLaunchDimensions() local
95 num_elements, threads_per_block, block_count); in CalculateLaunchDimensions()
97 return LaunchDimensions(block_count, threads_per_block); in CalculateLaunchDimensions()
/external/autotest/client/site_tests/platform_CrosDisksFilesystem/
Dvfat_tests5 "block_count": 65536,
15 "block_count": 65536,
25 "block_count": 65536,
37 "block_count": 65536,
Dexfat_tests5 "block_count": 65536,
15 "block_count": 65536,
25 "block_count": 65536,
Dext2_tests5 "block_count": 65536,
15 "block_count": 65536,
26 "block_count": 65536,
Dext3_tests5 "block_count": 65536,
15 "block_count": 65536,
26 "block_count": 65536,
Dext4_tests5 "block_count": 65536,
15 "block_count": 65536,
26 "block_count": 65536,
Dntfs_tests5 "block_count": 65536,
16 "block_count": 65536,
28 "block_count": 65536,
/external/tensorflow/tensorflow/core/kernels/
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()
219 <<<config.block_count, config.thread_per_block, shared_memory_size, in compute()
224 int group_size = (config.block_count + bias_size - 1) / bias_size; in compute()
225 config.block_count = group_size * bias_size; in compute()
230 BiasGradNCHW_SharedAtomics<T>, config.block_count, in compute()
240 BiasGradNHWC_Naive<T>, config.block_count, config.thread_per_block, 0, in compute()
243 TF_CHECK_OK(CudaLaunchKernel(BiasGradNCHW_Naive<T>, config.block_count, in compute()
Ddepthtospace_op_gpu.cu.cc166 D2S_NHWC<T>, config.block_count, config.thread_per_block, 0, d.stream(), in operator ()()
202 D2S_NCHW_LOOP<T, 2>, config.block_count, config.thread_per_block, in operator ()()
209 D2S_NCHW_LOOP<T, 3>, config.block_count, config.thread_per_block, in operator ()()
216 D2S_NCHW_LOOP<T, 4>, config.block_count, config.thread_per_block, in operator ()()
231 D2S_NCHW<T>, config.block_count, config.thread_per_block, 0, d.stream(), in operator ()()
Dinplace_ops_functor_gpu.cu.cc53 DoParallelConcatOpKernel<T>, cfg.block_count, cfg.thread_per_block, 0, in DoParallelConcatUpdate()
121 cfg.block_count, cfg.thread_per_block, 0, 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()
155 cfg.block_count, cfg.thread_per_block, 0, in DoInplaceOp()
Dspacetodepth_op_gpu.cu.cc162 S2D_NHWC<T>, config.block_count, config.thread_per_block, 0, d.stream(), in operator ()()
198 S2D_NCHW_LOOP<T, 2>, config.block_count, config.thread_per_block, in operator ()()
205 S2D_NCHW_LOOP<T, 3>, config.block_count, config.thread_per_block, in operator ()()
212 S2D_NCHW_LOOP<T, 4>, config.block_count, config.thread_per_block, in operator ()()
227 S2D_NCHW<T>, config.block_count, config.thread_per_block, 0, d.stream(), in operator ()()
Dresize_nearest_neighbor_op_gpu.cu.cc179 ResizeNearestNeighborNHWC<T>, config.block_count, in operator ()()
186 LegacyResizeNearestNeighborNHWC<T, align_corners>, config.block_count, in operator ()()
223 SetZero<<<output_config.block_count, output_config.thread_per_block, 0, in operator ()()
233 ResizeNearestNeighborBackwardNHWC<T>, input_config.block_count, in operator ()()
242 input_config.block_count, input_config.thread_per_block, 0, in operator ()()
Dadjust_saturation_op_gpu.cu.cc37 const int block_count = in operator ()() local
40 block_count, threads_per_block, 0, stream, in operator ()()
Dadjust_hue_op_gpu.cu.cc36 const int block_count = in operator ()() local
39 block_count, threads_per_block, 0, stream, in operator ()()
Dsegment_reduction_ops_gpu.cu.cc141 TF_CHECK_OK(CudaLaunchKernel(SetZero<T>, config.block_count, in operator ()()
168 config.block_count, config.thread_per_block, 0, d.stream(), in operator ()()
188 SetToValue<T>, config.block_count, config.thread_per_block, 0, in operator ()()
203 UnsortedSegmentCustomKernel<T, Index, ReductionF>, config.block_count, in operator ()()
Dcwise_op_clip_gpu.cu.cc66 UnaryClipCustomKernel<T>, config.block_count, config.thread_per_block, in operator ()()
82 BinaryRightClipCustomKernel<T>, config.block_count, in operator ()()
98 BinaryLeftClipCustomKernel<T>, config.block_count, in operator ()()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceThreadPool.h207 Index block_count = divup(n, block_size); in parallelFor() local
211 static_cast<double>(block_count) / in parallelFor()
212 (divup<int>(block_count, numThreads()) * numThreads()); in parallelFor()
215 for (Index prev_block_count = block_count; prev_block_count > 1;) { in parallelFor()
237 block_count = coarser_block_count; in parallelFor()
247 Barrier barrier(static_cast<unsigned int>(block_count)); in parallelFor()
/external/python/cpython2/Mac/Demo/applescript/Disk_Copy/
DStandard_Suite.py222 class block_count(aetools.NProperty): class
285 block_count = block_count variable
346 'block_count' : block_count,
373 'block_count' : block_count,
459 'Xblk' : block_count,
/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc152 int64 *block_count) { in CalculateDimensionality() argument
154 *block_count = port::MathUtil::CeilOfRatio(element_count, *threads_per_block); in CalculateDimensionality()
155 if (*block_count == 1) { in CalculateDimensionality()
/external/u-boot/disk/
Dpart_mac.c130 bytes = mpart->block_count; in part_print_mac()
144 mpart->block_count, in part_print_mac()
233 info->size = mpart->block_count; in part_get_info_mac()

12345