Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dkernel_mapping_scheme.h116 return CeilOfRatio(dims_in_elems_[0], GetTileSizeZ()) * in GetNumberOfBlocks()
125 int64 GetTileSizeZ() const { return GetTileSizeFor(DimZ); } in GetTileSizeZ() function
Dir_emitter_unnested.cc4699 CeilOfRatio(dims_in_elems[0], mapping_scheme.GetTileSizeZ()), in EmitTilingKernel()
4724 b_.CreateMul(starting_block[0], constant(mapping_scheme.GetTileSizeZ()), in EmitTilingKernel()
4761 if (mapping_scheme.GetTileSizeZ() == 1) { in EmitTilingKernel()
4765 llvm::Value* block_size_for_dim = constant(mapping_scheme.GetTileSizeZ()); in EmitTilingKernel()
4771 (dims_in_blocks[kDimZ] - 1) * mapping_scheme.GetTileSizeZ()); in EmitTilingKernel()
4942 bool block_contains_multi_tiles = mapping_scheme.GetTileSizeZ() > 1; in EmitHlo021Tile()