HomeSort by relevance Sort by last modified time
    Searched refs:MakeCollectiveAdapter (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
base_collective_executor.h 87 CollectiveAdapter* MakeCollectiveAdapter(Tensor* output, int num_chunks,
ring_gatherer.cc 92 ca_.reset(MakeCollectiveAdapter(col_ctx_->output, group_size_ * num_subdivs_,
ring_reducer.cc 113 ca_.reset(MakeCollectiveAdapter(col_ctx_->output, group_size_ * num_subdivs_,
base_collective_executor.cc 184 CollectiveAdapter* MakeCollectiveAdapter(Tensor* output, int num_chunks,
206 << " to MakeCollectiveAdapter";

Completed in 398 milliseconds