Home
last modified time | relevance | path

Searched refs:MakeCollectiveAdapter (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dbase_collective_executor.h87 CollectiveAdapter* MakeCollectiveAdapter(Tensor* output, int num_chunks,
Dring_gatherer.cc92 ca_.reset(MakeCollectiveAdapter(col_ctx_->output, group_size_ * num_subdivs_, in Run()
Dring_reducer.cc113 ca_.reset(MakeCollectiveAdapter(col_ctx_->output, group_size_ * num_subdivs_, in ContinueAfterInputCopy()
Dbase_collective_executor.cc184 CollectiveAdapter* MakeCollectiveAdapter(Tensor* output, int num_chunks, in MakeCollectiveAdapter() function