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.cc119 const size_t num_reduces = 32; in TEST_F() local
/external/pytorch/torch/csrc/distributed/c10d/
DProcessGroupNCCL.cpp3811 const auto num_reduces = outputTensors_.size(); in checkForNCCLErrorsInternal() local
3947 const auto num_reduces = inputTensors_.size(); in checkForNCCLErrorsInternal() local