Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dkernel_mapping_scheme.h126 int64_t GetNumThreadsPerBlockPhysical() const { in GetNumThreadsPerBlockPhysical() function
H A Dir_emitter_unnested.cc3866 tiling_scheme.GetNumThreadsPerBlockPhysical(), num_rows_per_warp); in EmitReductionOutputForRowReduction()
4072 EmitThreadId(tiling_scheme.GetNumThreadsPerBlockPhysical(), index_ty); in EmitThreadIdInfo()
4327 tiling_scheme.GetNumThreadsPerBlockPhysical()); in Emit021Transpose()
4708 CHECK_EQ(tiling_scheme.GetNumThreadsPerBlockPhysical() % WarpSize(), 0); in EmitIRForReduction()
4711 tiling_scheme.GetNumThreadsPerBlockPhysical() * in EmitIRForReduction()
4935 {/*x=*/tiling_scheme.GetNumThreadsPerBlockPhysical(), /*y=*/1, /*z=*/1}); in EmitUnnestedReduction()