Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.cc138 int64 tile_z = std::min(reduction_dimensions.dimensions[0], int64{8}); in GetReductionTiling() local
141 return {tile_z, 1, 16}; in GetReductionTiling()
145 return {tile_z, 1, 64}; in GetReductionTiling()
157 return {tile_z, 1, unroll_x}; in GetReductionTiling()