Home
last modified time | relevance | path

Searched refs:buffer_assignment_ (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_context.h46 buffer_assignment_(buffer_assignment), in IrEmitterContext()
60 return *buffer_assignment_; in buffer_assignment()
77 if (buffer_assignment_) { in allocations()
78 return buffer_assignment_->Allocations(); in allocations()
84 CHECK_EQ(nullptr, buffer_assignment_); in set_allocations()
90 const BufferAssignment* buffer_assignment_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_executable.cc66 buffer_assignment_.reset(new BufferAssignmentProto(assignment_->ToProto())); in CpuExecutable()
70 module().unique_id(), shared_module(), buffer_assignment_); in CpuExecutable()
H A Dcpu_executable.h132 std::shared_ptr<const BufferAssignmentProto> buffer_assignment_; variable