Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3657 int64 block_size_z = 1; in ComputeMappingSchemeAndReductionKind() local
3710 block_size_z = 8; in ComputeMappingSchemeAndReductionKind()
3711 while (depth % block_size_z != 0) { in ComputeMappingSchemeAndReductionKind()
3712 block_size_z -= 1; in ComputeMappingSchemeAndReductionKind()
3721 DimensionVector req_block_sizes{block_size_z, 1, 1}; in ComputeMappingSchemeAndReductionKind()