Searched refs:DimX (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | kernel_tiling.h | 90 enum { DimZ = 0, DimY, DimX, DimTot }; enumerator
121 DCHECK(d >= DimZ && d <= DimX); in GetNumberOfTilesInOneBlockForDimension()
129 DCHECK(d >= DimZ && d <= DimX); in GetTileSizeForDimension()
133 return GetTileSizeForDimension(DimX); in GetTileSizeForDimensionX()
141 DCHECK(d >= DimZ && d <= DimX); in GetTileBlockSizeForDimension()
159 CHECK_EQ(GetTileSizeForDimension(DimX) % num_threads_x_, 0); in SetDilatedX()
|
| D | kernel_tiling.cc | 211 llvm::ArrayType::get(elem_ty, GetTileSizeForDimension(DimX) + 1), in GetSharedMemoryBufferForElementType()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2332 .AddOffsetToDim(start_offset_x, KernelMappingScheme::DimX, builder); in EmitFullElementalTile()
2344 KernelMappingScheme::DimX, builder); in EmitFullElementalTile()
2371 .AddOffsetToDim(start_offset_x, KernelMappingScheme::DimX, builder); in EmitPartialElementalTile()
2375 KernelMappingScheme::DimX, builder); in EmitPartialElementalTile()
2579 return IsRowReduction() ? llvm_ir::KernelMappingScheme::DimX in GetReducedDimensionEnum()
2586 : llvm_ir::KernelMappingScheme::DimX; in GetKeptDimensionEnum()
3032 KernelMappingScheme::DimX, in EmitBlock()
|