Searched defs:buffer_assignment (Results 1 – 15 of 15) sorted by relevance
46 const xla::BufferAssignment& buffer_assignment) { in ComputeResultIndex()128 const xla::BufferAssignment& buffer_assignment = in Compile() local
41 const HloModule* hlo_module, const BufferAssignment* buffer_assignment, in IrEmitterContext()60 const BufferAssignment& buffer_assignment() const { in buffer_assignment() function
38 const BufferAssignment* buffer_assignment, in HloToIrBindings()
556 std::unique_ptr<BufferAssignment>* buffer_assignment, in CompileModuleToLlvmIrImpl()900 std::unique_ptr<BufferAssignment> buffer_assignment; in RunBackend() local990 std::unique_ptr<BufferAssignment> buffer_assignment; in CompileModuleToLlvmIr() local
173 const BufferAssignment* buffer_assignment) { in BuffersInvariantWithinConsumer()
1748 const auto& buffer_assignment = ir_emitter_context_->buffer_assignment(); in GetMlirEmitterInput() local3760 const BufferAssignment& buffer_assignment) { in CheckHloBuffersShareAllocation()3779 const BufferAssignment& buffer_assignment) { in CheckWhileBuffersShareAllocation()3808 const BufferAssignment& buffer_assignment) { in CheckConditionalBuffersShareAllocation()
24 std::shared_ptr<const BufferAssignmentProto> buffer_assignment) { in RegisterModule()42 std::shared_ptr<const BufferAssignmentProto> buffer_assignment) { in UnregisterModule()
115 std::shared_ptr<const BufferAssignmentProto> buffer_assignment; member
32 std::shared_ptr<BufferAssignmentProto> buffer_assignment; member
512 BufferAssignmentProto buffer_assignment = 3; field
25 const BufferAssignment& buffer_assignment) { in CreateBufferInfosFromBufferAssignment()
82 const BufferAssignment& buffer_assignment() const { return *assignment_; } in buffer_assignment() function
600 ObjectsAssignment<size_t> buffer_assignment; in AssignInternalObjects() local
545 ObjectsAssignment<size_t> buffer_assignment; in AllocateMemoryForBuffers() local
498 ObjectsAssignment<size_t> buffer_assignment; in AllocateMemoryForBuffers() local