Searched refs:DimY (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
136 return GetTileSizeForDimension(DimY); in GetTileSizeForDimensionY()
|
| D | kernel_tiling.cc | 189 for (int i = DimY; i < DimTot; ++i) { in GetElementIndexForTileOrigin()
212 GetTileSizeForDimension(DimY)); in GetSharedMemoryBufferForElementType()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2331 tile_origin_index.AddOffsetToDim(y, KernelMappingScheme::DimY, builder) in EmitFullElementalTile()
2338 y_indvar, KernelMappingScheme::DimY, builder); in EmitFullElementalTile()
2370 tile_origin_index.AddOffsetToDim(y, KernelMappingScheme::DimY, builder) in EmitPartialElementalTile()
2401 y_indvar, KernelMappingScheme::DimY, builder), in EmitPartialElementalTile()
2580 : llvm_ir::KernelMappingScheme::DimY; in GetReducedDimensionEnum()
2585 return IsRowReduction() ? llvm_ir::KernelMappingScheme::DimY in GetKeptDimensionEnum()
3001 for (int i = KernelMappingScheme::DimY; i < KernelMappingScheme::DimTot; in EmitBlock()
3029 "block_dim_y", starting_tile_for_dim_y, KernelMappingScheme::DimY, in EmitBlock()
|