Home
last modified time | relevance | path

Searched refs:GetTileSizeForDimensionY (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_tiling.h135 int64 GetTileSizeForDimensionY() const { in GetTileSizeForDimensionY() function
Dkernel_tiling.cc202 CHECK_EQ(GetTileSizeForDimensionX(), GetTileSizeForDimensionY()); in GetSharedMemoryBufferForElementType()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2324 int64 tile_size_y = mapping_scheme->GetTileSizeForDimensionY(); in EmitFullElementalTile()
2424 int64 tile_size_y = mapping_scheme->GetTileSizeForDimensionY(); in EmitTiledElementalCodeWithBoundsCheck()