HomeSort by relevance Sort by last modified time
    Searched refs:replicated_buffers (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
allocation_tracker.cc 37 std::vector<ScopedShapedBuffer> replicated_buffers; local
38 replicated_buffers.emplace_back(std::move(shaped_buffer));
39 return RegisterInternal(std::move(replicated_buffers), tag);
43 std::vector<ScopedShapedBuffer> replicated_buffers, const string& tag) {
46 return RegisterInternal(std::move(replicated_buffers), tag);
59 std::vector<ShapedBufferTy> replicated_buffers, const string& tag) {
64 << "tag: \"" << tag << "\" with " << replicated_buffers.size()
66 for (const auto& shaped_buffer : replicated_buffers) {
77 for (auto& shaped_buffer : replicated_buffers) {
108 TF_ASSIGN_OR_RETURN(std::vector<const ShapedBuffer*> replicated_buffers,
167 std::vector<ShapedBuffer> replicated_buffers; local
206 std::vector<const ShapedBuffer*> replicated_buffers; local
    [all...]
local_service.h 69 std::vector<ScopedShapedBuffer> replicated_buffers, const string& tag);
allocation_tracker.h 55 std::vector<ScopedShapedBuffer> replicated_buffers, const string& tag);
98 std::vector<ShapedBufferTy> replicated_buffers, const string& tag)
local_service.cc 201 std::vector<ScopedShapedBuffer> replicated_buffers, const string& tag) {
203 std::move(replicated_buffers), tag);
service.cc 243 auto replicated_buffers = buffer_status.ValueOrDie(); local
244 CHECK_EQ(options_.number_of_replicas(), replicated_buffers.size());
246 const ShapedBuffer* shaped_buffer = replicated_buffers[replica];
987 std::vector<ScopedShapedBuffer> replicated_buffers; local
    [all...]

Completed in 94 milliseconds