Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_context.h47 buffer_assignment_(buffer_assignment), in IrEmitterContext()
61 return *buffer_assignment_; in buffer_assignment()
76 if (buffer_assignment_) { in allocations()
77 return buffer_assignment_->Allocations(); in allocations()
83 CHECK_EQ(nullptr, buffer_assignment_); in set_allocations()
89 const BufferAssignment* buffer_assignment_; variable
Dhlo_to_ir_bindings.h41 : buffer_assignment_(buffer_assignment), in HloToIrBindings()
101 const BufferAssignment* buffer_assignment_; variable
Dhlo_to_ir_bindings.cc207 BuffersInvariantWithinConsumer(hlo, consumer, buffer_assignment_)) { in GetIrArray()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc67 buffer_assignment_.reset(new BufferAssignmentProto(assignment_->ToProto())); in CpuExecutable()
70 buffer_assignment_); in CpuExecutable()
88 buffer_assignment_); in ~CpuExecutable()
Dcpu_executable.h134 std::shared_ptr<const BufferAssignmentProto> buffer_assignment_; variable