| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| partition_assignment.h | 41 LaunchDimensions(int64 block_count, int64 threads_per_block)
42 : block_count_(block_count), threads_per_block_(threads_per_block) {}
48 int64 block_count() const { return block_count_; } function in class:xla::gpu::LaunchDimensions
|
| partition_assignment.cc | 40 launch_dims.block_count(),
91 int64 block_count = CeilOfRatio(num_elements, threads_per_block); local
95 num_elements, threads_per_block, block_count);
97 return LaunchDimensions(block_count, threads_per_block);
|
| parallel_loop_emitter.cc | 68 llvm_ir::AddRangeMetadata(0, launch_dimensions_.block_count(),
104 launch_dimensions_.block_count()),
|
| /external/tensorflow/tensorflow/core/kernels/ |
| adjust_hue_op_gpu.cu.cc | 34 const int block_count = local
37 <<<block_count, threads_per_block, 0, stream>>>(
|
| adjust_saturation_op_gpu.cu.cc | 36 const int block_count = local
39 <<<block_count, threads_per_block, 0, stream>>>(
|
| bias_op_gpu.cu.cc | 81 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
85 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
210 <<<config.block_count, config.thread_per_block, shared_memory_size,
215 int group_size = (config.block_count + bias_size - 1) / bias_size;
216 config.block_count = group_size * bias_size;
221 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
231 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
235 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
|
| fused_batch_norm_op.cu.cc | 41 VarianceToInvVarianceKernel<<<config.block_count, config.thread_per_block, 0,
63 InvVarianceToVarianceKernel<<<config.block_count, config.thread_per_block, 0,
|
| depthtospace_op_gpu.cu.cc | 162 D2S_NHWC<<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
195 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
201 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
207 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
217 D2S_NCHW<<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
|
| spacetodepth_op_gpu.cu.cc | 158 S2D_NHWC<<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
191 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
197 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
203 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
213 S2D_NCHW<<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
|
| bucketize_op_gpu.cu.cc | 102 <<<config.block_count, config.thread_per_block, shared_mem_size,
107 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
|
| diag_op_gpu.cu.cc | 64 DiagCudaKernel<<<diag_config.block_count, diag_config.thread_per_block, 0,
104 DiagPartCudaKernel<<<diag_config.block_count, diag_config.thread_per_block,
|
| gather_functor_gpu.cu.h | 97 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
104 <<<config.block_count, config.thread_per_block, 0, d.stream()>>>(
|
| matrix_set_diag_op_gpu.cu.cc | 77 <<<config.block_count, config.thread_per_block, 0, device.stream()>>>(
83 <<<config.block_count, config.thread_per_block, 0, device.stream()>>>(
|
| /external/autotest/client/site_tests/platform_DMVerityCorruption/ |
| platform_DMVerityCorruption.py | 14 block_count):
16 run_count, backing_path, block_size, block_count))
18 run_count = run_count % block_count
22 block_count):
24 run_count, backing_path, block_size, block_count))
26 dev.seek(block_count * block_size, os.SEEK_SET)
|
| /external/tensorflow/tensorflow/core/util/ |
| cuda_launch_config.h | 77 MyKernel1D <<<config.block_count,
80 MyKernel2D <<<config.block_count,
83 MyKernel3D <<<config.block_count,
88 MyKernel1D <<<config.block_count,
92 MyKernel2D <<<config.block_count,
96 MyKernel3D <<<config.block_count,
118 int block_count = -1; member in struct:tensorflow::CudaLaunchConfig
134 const int block_count = local
140 config.block_count = block_count;
155 int block_count = 0; local
182 int block_count = 0; local
199 dim3 block_count = dim3(0, 0, 0); member in struct:tensorflow::Cuda2DLaunchConfig
256 int block_count = 0; local
[all...] |
| cuda_kernel_helper_test.cu.cc | 155 SetOutbufZero<<<cfg.block_count, cfg.thread_per_block, 0, d.stream()>>>( \
159 Count1D<<<cfg.block_count, cfg.thread_per_block, 0, d.stream()>>>( \
165 SetOutbufZero<<<cfg.block_count, cfg.thread_per_block, 0, d.stream()>>>( \
169 Count1D<<<cfg.block_count, cfg.thread_per_block, 0, d.stream()>>>( \
191 a.block_count.x == b.block_count.x &&
192 a.block_count.y == b.block_count.y &&
193 a.block_count.z == b.block_count.z &
[all...] |
| /external/autotest/client/site_tests/platform_DMVerityBitCorruption/ |
| platform_DMVerityBitCorruption.py | 18 block_count):
20 run_count, backing_path, block_size, block_count))
21 run_count = run_count % block_count
30 block_count):
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)
|
| /system/update_engine/payload_generator/ |
| raw_filesystem.cc | 28 const std::string& filename, uint64_t block_size, uint64_t block_count) {
32 result->block_count_ = block_count;
|
| fake_filesystem.cc | 23 FakeFilesystem::FakeFilesystem(uint64_t block_size, uint64_t block_count) :
25 block_count_(block_count) {
|
| fake_filesystem.h | 34 FakeFilesystem(uint64_t block_size, uint64_t block_count);
|
| raw_filesystem.h | 33 const std::string& filename, uint64_t block_size, uint64_t block_count);
|
| extent_utils.cc | 119 uint64_t block_offset, uint64_t block_count) {
122 if (block_count == 0)
124 uint64_t end_block_offset = block_offset + block_count;
128 // of the loop scanned_blocks < block_offset + block_count.
|
| /external/fio/os/ |
| os-mac.h | 53 uint64_t block_count; local
55 if (ioctl(f->fd, DKIOCGETBLOCKCOUNT, &block_count) == -1)
61 *bytes *= block_count;
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorDeviceThreadPool.h | 207 Index block_count = divup(n, block_size); local
211 static_cast<double>(block_count) /
212 (divup<int>(block_count, numThreads()) * numThreads());
215 for (Index prev_block_count = block_count; prev_block_count > 1;) {
237 block_count = coarser_block_count;
246 // block_count leaves that do actual computations.
247 Barrier barrier(static_cast<unsigned int>(block_count));
|
| /external/python/cpython2/Mac/Demo/applescript/Disk_Copy/ |
| Standard_Suite.py | 222 class block_count(aetools.NProperty): class in inherits:aetools.NProperty
285 block_count = block_count variable
346 'block_count' : block_count,
373 'block_count' : block_count,
459 'Xblk' : block_count,
|
