Searched refs:CollectiveAdapterImpl (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 90 class CollectiveAdapterImpl : public CollectiveAdapter { class
94 CollectiveAdapterImpl(Tensor* output, int64_t num_chunks, in CollectiveAdapterImpl() function in tensorflow::__anonb82aea130211::CollectiveAdapterImpl
114 ~CollectiveAdapterImpl() override {} in ~CollectiveAdapterImpl()
200 return new CollectiveAdapterImpl<Eigen::bfloat16>( in MakeCollectiveAdapter()
204 return new CollectiveAdapterImpl<Eigen::half>(output, num_chunks, in MakeCollectiveAdapter()
208 return new CollectiveAdapterImpl<float>(output, num_chunks, allocator, in MakeCollectiveAdapter()
212 return new CollectiveAdapterImpl<double>(output, num_chunks, allocator, in MakeCollectiveAdapter()
216 return new CollectiveAdapterImpl<int32>(output, num_chunks, allocator, in MakeCollectiveAdapter()
220 return new CollectiveAdapterImpl<int64_t>(output, num_chunks, allocator, in MakeCollectiveAdapter()
|