Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_tiling.h128 int64 GetTileSizeForDimension(int d) const { in GetTileSizeForDimension() function
133 return GetTileSizeForDimension(DimX); in GetTileSizeForDimensionX()
136 return GetTileSizeForDimension(DimY); in GetTileSizeForDimensionY()
159 CHECK_EQ(GetTileSizeForDimension(DimX) % num_threads_x_, 0); in SetDilatedX()
Dkernel_tiling.cc193 GetTileSizeForDimension(i)), in GetElementIndexForTileOrigin()
211 llvm::ArrayType::get(elem_ty, GetTileSizeForDimension(DimX) + 1), in GetSharedMemoryBufferForElementType()
212 GetTileSizeForDimension(DimY)); in GetSharedMemoryBufferForElementType()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3003 int64 tile_size_for_dim = mapping_scheme->GetTileSizeForDimension(i); in EmitBlock()