Searched defs:threads_per_block (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | partition_assignment.h | 41 LaunchDimensions(int64 block_count, int64 threads_per_block) in LaunchDimensions()
49 int64 threads_per_block() const { return threads_per_block_; } in threads_per_block() function
|
| D | partition_assignment.cc | 43 int64 threads_per_block = device_desc.threads_per_block_limit(); in ThreadsPerBlockLimit() local
84 int64 threads_per_block = ThreadsPerBlockLimit(device_desc); in CalculateLaunchDimensions() local
|
| D | elemental_ir_emitter.cc | 321 llvm::Value* threads_per_block = in EmitThreadId() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | adjust_saturation_op_gpu.cu.cc | 36 const int threads_per_block = config.thread_per_block; in operator ()() local
|
| D | adjust_hue_op_gpu.cu.cc | 35 const int threads_per_block = config.thread_per_block; in operator ()() local
|
| /external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/ |
| D | inc_op_kernel.cu.cc | 39 int threads_per_block = 256; in IncrementKernel() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_description.cc | 151 int64 element_count, int64 *threads_per_block, in CalculateDimensionality()
|
| /external/mesa3d/src/gallium/drivers/radeonsi/ |
| D | si_get.c | 785 unsigned threads_per_block = get_max_threads_per_block(sscreen, ir_type); in si_get_compute_param() local
|
| /external/mesa3d/src/gallium/drivers/r600/ |
| D | r600_pipe_common.c | 1041 unsigned threads_per_block = get_max_threads_per_block(rscreen, ir_type); in r600_get_compute_param() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 1356 GpuContext* context, hipFunction_t kernel, int threads_per_block, in GetMaxOccupiedBlocksPerCore()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.cc | 1613 GpuContext* context, CUfunction kernel, int threads_per_block, in GetMaxOccupiedBlocksPerCore()
|