Searched refs:CollectiveAdapterImpl (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 88 class CollectiveAdapterImpl : public CollectiveAdapter { class
92 CollectiveAdapterImpl(Tensor* output, int64 num_chunks, Allocator* allocator, in CollectiveAdapterImpl() function in tensorflow::__anon2cb66ace0211::CollectiveAdapterImpl
112 ~CollectiveAdapterImpl() override {} in ~CollectiveAdapterImpl()
198 return new CollectiveAdapterImpl<Eigen::half>(output, num_chunks, in MakeCollectiveAdapter()
202 return new CollectiveAdapterImpl<float>(output, num_chunks, allocator, in MakeCollectiveAdapter()
206 return new CollectiveAdapterImpl<double>(output, num_chunks, allocator, in MakeCollectiveAdapter()
210 return new CollectiveAdapterImpl<int32>(output, num_chunks, allocator, in MakeCollectiveAdapter()
214 return new CollectiveAdapterImpl<int64>(output, num_chunks, allocator, in MakeCollectiveAdapter()
|