Home
last modified time | relevance | path

Searched refs:block_count (Results 1 – 25 of 107) 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.cc155 SetOutbufZero<<<cfg.block_count, cfg.thread_per_block, 0, d.stream()>>>( \ in TEST_F()
159 Count1D<<<cfg.block_count, cfg.thread_per_block, 0, d.stream()>>>( \ in TEST_F()
165 SetOutbufZero<<<cfg.block_count, cfg.thread_per_block, 0, d.stream()>>>( \ in TEST_F()
169 Count1D<<<cfg.block_count, cfg.thread_per_block, 0, d.stream()>>>( \ in TEST_F()
191 a.block_count.x == b.block_count.x && in operator ==()
192 a.block_count.y == b.block_count.y && in operator ==()
193 a.block_count.z == b.block_count.z && in operator ==()
206 SetOutbufZero<<<cfg1d.block_count, cfg1d.thread_per_block, 0, d.stream()>>>( \ in TEST_F()
210 Count2D<<<cfg.block_count, cfg.thread_per_block, 0, d.stream()>>>( \ in TEST_F()
216 SetOutbufZero<<<cfg1d.block_count, cfg1d.thread_per_block, 0, d.stream()>>>( \ 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.cc40 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()
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
/external/tensorflow/tensorflow/core/kernels/
Dbias_op_gpu.cu.cc81 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in compute()
85 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in compute()
210 <<<config.block_count, config.thread_per_block, shared_memory_size, in compute()
215 int group_size = (config.block_count + bias_size - 1) / bias_size; in compute()
216 config.block_count = group_size * bias_size; in compute()
221 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in compute()
231 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in compute()
235 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in compute()
Dspacetodepth_op_gpu.cu.cc158 S2D_NHWC<<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
191 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
197 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
203 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
213 S2D_NCHW<<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
Ddepthtospace_op_gpu.cu.cc162 D2S_NHWC<<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
195 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
201 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
207 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
217 D2S_NCHW<<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
Dadjust_saturation_op_gpu.cu.cc36 const int block_count = in operator ()() local
39 <<<block_count, threads_per_block, 0, stream>>>( in operator ()()
Dadjust_hue_op_gpu.cu.cc34 const int block_count = in operator ()() local
37 <<<block_count, threads_per_block, 0, stream>>>( in operator ()()
Dsegment_reduction_ops_gpu.cu.cc138 SetZero<<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
163 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
182 SetToValue<<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
197 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
Ddilation_ops_gpu.cu.cc199 DilationKernel<<<config.block_count, config.thread_per_block, 0, in operator ()()
233 SetZero<<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
239 DilationBackpropInputKernel<<<config.block_count, config.thread_per_block, in operator ()()
273 SetZero<<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
279 DilationBackpropFilterKernel<<<config.block_count, config.thread_per_block, in operator ()()
Dsplit_lib_gpu.cu.cc200 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in Run()
216 SplitVOpKernel_fixed<T><<<config.block_count, config.thread_per_block, 0, in Run()
229 <<<config.block_count, config.thread_per_block, smem_usage, in Run()
234 <<<config.block_count, config.thread_per_block, 0, in Run()
/external/autotest/client/site_tests/platform_CrosDisksFilesystem/
Dexfat_tests5 "block_count": 65536,
15 "block_count": 65536,
25 "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,
Dext2_tests5 "block_count": 65536,
15 "block_count": 65536,
26 "block_count": 65536,
Dvfat_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/fio/os/
Dos-mac.h53 uint64_t block_count; in blockdev_size() local
55 if (ioctl(f->fd, DKIOCGETBLOCKCOUNT, &block_count) == -1) in blockdev_size()
61 *bytes *= block_count; in blockdev_size()
/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/libjpeg-turbo/
Drdtarga.c70 int block_count; /* # of pixels remaining in RLE block */ member
151 if (--sinfo->block_count < 0) { /* decrement pixels remaining in block */ in read_rle_pixel()
155 sinfo->block_count = 0; /* then read new block header */ in read_rle_pixel()
157 sinfo->block_count = i & 0x7F; /* number of pixels after this one */ in read_rle_pixel()
375 source->block_count = source->dup_pixel_count = 0; in start_input_tga()
/external/tensorflow/tensorflow/contrib/resampler/kernels/
Dresampler_ops_gpu.cu.cc123 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
258 SetZero<<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
263 SetZero<<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()
270 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>( in operator ()()

12345