Searched refs:num_threads_x (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | kernel_tiling.cc | 121 int64 num_threads_x, llvm::IRBuilder<>* b) in KernelMappingScheme() argument
125 num_threads_x_(num_threads_x),
|
| D | kernel_tiling.h | 101 int64 num_threads_y, int64 num_threads_x,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2299 int64 tile_size_x, int64 num_threads_x, in GetStartOffsetAndStepForX() argument
2306 step_x = num_threads_x; in GetStartOffsetAndStepForX()
2309 x, llvm::ConstantInt::get(index_ty, tile_size_x / num_threads_x)); in GetStartOffsetAndStepForX()
2321 int64 num_threads_x = mapping_scheme->GetNumberOfThreadsForDimensionX(); in EmitFullElementalTile() local
2329 tile_size_x, num_threads_x, mapping_scheme, builder, x, index_ty); in EmitFullElementalTile()
2341 for (int64 j = 0; j < tile_size_x / num_threads_x; j++) { in EmitFullElementalTile()
2361 int64 num_threads_x = mapping_scheme->GetNumberOfThreadsForDimensionX(); in EmitPartialElementalTile() local
2368 tile_size_x, num_threads_x, mapping_scheme, builder, x, index_ty); in EmitPartialElementalTile()
2372 for (int64 j = 0; j < tile_size_x / num_threads_x; j++) { in EmitPartialElementalTile()
3658 int64 num_threads_x = 1; in ComputeMappingSchemeAndReductionKind() local
[all …]
|