Searched refs:CollectiveAdapterImpl (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 77 class CollectiveAdapterImpl : public CollectiveAdapter { class
81 CollectiveAdapterImpl(Tensor* output, int64 num_chunks, Allocator* allocator, in CollectiveAdapterImpl() function in tensorflow::__anon2cb66ace0111::CollectiveAdapterImpl
101 ~CollectiveAdapterImpl() override {} in ~CollectiveAdapterImpl()
189 return new CollectiveAdapterImpl<float>(output, num_chunks, allocator, in MakeCollectiveAdapter()
193 return new CollectiveAdapterImpl<double>(output, num_chunks, allocator, in MakeCollectiveAdapter()
197 return new CollectiveAdapterImpl<int32>(output, num_chunks, allocator, in MakeCollectiveAdapter()
201 return new CollectiveAdapterImpl<int64>(output, num_chunks, allocator, in MakeCollectiveAdapter()
|