Home
last modified time | relevance | path

Searched defs:num_reduces (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dparallel_reduction_test.cc78 const size_t num_reduces = 32; in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc5478 HloComputation* fused_computation, int num_reduces) { in DivideOutputInstructionsIntoGroups()
5535 int num_reduces = 1; in EmitReductionFromOrToContiguousDimensions() local