Searched refs:replicated_buffers (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | allocation_tracker.cc | 37 std::vector<ScopedShapedBuffer> replicated_buffers; in Register() local
38 replicated_buffers.emplace_back(std::move(shaped_buffer)); in Register()
39 return RegisterInternal(std::move(replicated_buffers), tag); in Register()
43 std::vector<ScopedShapedBuffer> replicated_buffers, const string& tag) { in RegisterReplicatedBuffers() argument
46 return RegisterInternal(std::move(replicated_buffers), tag); in RegisterReplicatedBuffers()
59 std::vector<ShapedBufferTy> replicated_buffers, const string& tag) { in RegisterInternal() argument
64 << "tag: \"" << tag << "\" with " << replicated_buffers.size() in RegisterInternal()
66 for (const auto& shaped_buffer : replicated_buffers) { in RegisterInternal()
77 for (auto& shaped_buffer : replicated_buffers) { in RegisterInternal()
108 TF_ASSIGN_OR_RETURN(std::vector<const ShapedBuffer*> replicated_buffers, in Unregister()
[all …]
|
| D | allocation_tracker.h | 55 std::vector<ScopedShapedBuffer> replicated_buffers, const string& tag);
98 std::vector<ShapedBufferTy> replicated_buffers, const string& tag)
|
| D | local_service.cc | 201 std::vector<ScopedShapedBuffer> replicated_buffers, const string& tag) { in RegisterReplicatedBuffers() argument
203 std::move(replicated_buffers), tag); in RegisterReplicatedBuffers()
|
| D | local_service.h | 69 std::vector<ScopedShapedBuffer> replicated_buffers, const string& tag);
|
| D | service.cc | 243 auto replicated_buffers = buffer_status.ValueOrDie(); in ResolveAndValidateArguments() local
244 CHECK_EQ(options_.number_of_replicas(), replicated_buffers.size()); in ResolveAndValidateArguments()
246 const ShapedBuffer* shaped_buffer = replicated_buffers[replica]; in ResolveAndValidateArguments()
987 std::vector<ScopedShapedBuffer> replicated_buffers; in TransferToServer() local
998 replicated_buffers.emplace_back(std::move(shaped_buffer)); in TransferToServer()
1002 std::move(replicated_buffers), in TransferToServer()
|