Searched refs:replicated_buffers (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | allocation_tracker.cc | 38 std::vector<ScopedShapedBuffer> replicated_buffers; in Register() local
39 replicated_buffers.emplace_back(std::move(shaped_buffer)); in Register()
40 return RegisterInternal(std::move(replicated_buffers), tag); in Register()
44 std::vector<ScopedShapedBuffer> replicated_buffers, const string& tag) { in RegisterReplicatedBuffers() argument
47 return RegisterInternal(std::move(replicated_buffers), tag); in RegisterReplicatedBuffers()
60 std::vector<ShapedBufferTy> replicated_buffers, const string& tag) { in RegisterInternal() argument
65 << "tag: \"" << tag << "\" with " << replicated_buffers.size() in RegisterInternal()
69 for (auto& shaped_buffer : replicated_buffers) { in RegisterInternal()
100 TF_ASSIGN_OR_RETURN(std::vector<const ShapedBuffer*> replicated_buffers, in Unregister()
102 for (const auto& shaped_buffer : 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 | 208 std::vector<ScopedShapedBuffer> replicated_buffers, const string& tag) { in RegisterReplicatedBuffers() argument
210 std::move(replicated_buffers), tag); in RegisterReplicatedBuffers()
|
| D | local_service.h | 70 std::vector<ScopedShapedBuffer> replicated_buffers, const string& tag);
|
| D | service.cc | 245 auto replicated_buffers = buffer_status.ValueOrDie(); in ResolveAndValidateArguments() local
246 CHECK_EQ(options_.number_of_replicas(), replicated_buffers.size()); in ResolveAndValidateArguments()
248 const ShapedBuffer* shaped_buffer = replicated_buffers[replica]; in ResolveAndValidateArguments()
964 std::vector<ScopedShapedBuffer> replicated_buffers; in TransferToServer() local
975 replicated_buffers.emplace_back(std::move(shaped_buffer)); in TransferToServer()
979 std::move(replicated_buffers), in TransferToServer()
|