Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dlaunch_dimensions.cc59 int unroll_factor, bool few_waves) { in CalculateLaunchDimensions() argument
93 if (few_waves) { in CalculateLaunchDimensions()
Dlaunch_dimensions.h71 bool few_waves = false);
Dir_emitter_unnested.h235 KernelThunk* thunk, int unroll_factor, bool few_waves = false);
Dir_emitter_unnested.cc1820 bool few_waves = [fusion]() mutable { in EmitLoopFusionFromMlir() local
1843 few_waves); in EmitLoopFusionFromMlir()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dlaunch_dimensions_big.hlo2 // This test that we cap grid launch code when few_waves is enabled.
Dlaunch_dimensions.hlo3 // few_waves is enabled.