Searched refs:CollectiveAdapterImpl (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 79 class CollectiveAdapterImpl : public CollectiveAdapter { class
83 CollectiveAdapterImpl(Tensor* output, int64 num_chunks, Allocator* allocator, in CollectiveAdapterImpl() function in tensorflow::__anon509782990111::CollectiveAdapterImpl
103 ~CollectiveAdapterImpl() override {} in ~CollectiveAdapterImpl()
188 return new CollectiveAdapterImpl<Eigen::half>(output, num_chunks, in MakeCollectiveAdapter()
192 return new CollectiveAdapterImpl<float>(output, num_chunks, allocator, in MakeCollectiveAdapter()
196 return new CollectiveAdapterImpl<double>(output, num_chunks, allocator, in MakeCollectiveAdapter()
200 return new CollectiveAdapterImpl<int32>(output, num_chunks, allocator, in MakeCollectiveAdapter()
204 return new CollectiveAdapterImpl<int64>(output, num_chunks, allocator, in MakeCollectiveAdapter()
|