Home
last modified time | relevance | path

Searched refs:GetTileSizeY (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dkernel_mapping_scheme.h117 CeilOfRatio(dims_in_elems_[1], GetTileSizeY()) * in GetNumberOfBlocks()
127 int64 GetTileSizeY() const { return GetTileSizeFor(DimY); } in GetTileSizeY() function
Dir_emitter_unnested.cc4700 CeilOfRatio(dims_in_elems[1], mapping_scheme.GetTileSizeY()), in EmitTilingKernel()
4856 mapping_scheme.GetTileSizeY()); in EmitHlo021Tile()