Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dkernel_mapping_scheme.h132 int64 GetThreadsPerBlock() const { in GetThreadsPerBlock() function
Dir_emitter_unnested.cc4391 EmitThreadIdInfo(mapping_scheme.GetThreadsPerBlock(), index_ty, in EmitEpilogueForReduction()
4482 mapping_scheme.GetThreadsPerBlock()); in EmitEpilogueForReduction()
4513 mapping_scheme.GetThreadsPerBlock()); in EmitEpilogueForReduction()
4542 mapping_scheme.GetThreadsPerBlock()); in EmitEpilogueForReduction()
4707 EmitThreadIdInfo(mapping_scheme.GetThreadsPerBlock(), index_ty, in EmitTilingKernel()
4834 mapping_scheme.GetThreadsPerBlock()); in EmitHlo021Tile()
5429 mapping_scheme.GetThreadsPerBlock()); in EmitIRForReduction()
5684 {/*x=*/mapping_scheme.GetThreadsPerBlock(), /*y=*/1, /*z=*/1}); in EmitReductionFromOrToContiguousDimensions()
5688 << " - threads per block: " << mapping_scheme.GetThreadsPerBlock(); in EmitReductionFromOrToContiguousDimensions()