Searched defs:buffer_assignment (Results 1 – 12 of 12) sorted by relevance
46 const xla::BufferAssignment& buffer_assignment) { in ComputeResultIndex()128 const xla::BufferAssignment& buffer_assignment = in Compile() local
40 const BufferAssignment* buffer_assignment, in IrEmitterContext()59 const BufferAssignment& buffer_assignment() const { in buffer_assignment() function
953 std::unique_ptr<BufferAssignment> buffer_assignment; member1399 std::shared_ptr<const BufferAssignment> buffer_assignment( in RunBackend() local
31 std::shared_ptr<const BufferAssignmentProto> buffer_assignment) { in RegisterModule()77 std::shared_ptr<BufferAssignmentProto> buffer_assignment; member
68 std::shared_ptr<const BufferAssignmentProto> buffer_assignment; member
28 std::shared_ptr<const BufferAssignmentProto> buffer_assignment) { in RegisterModule()
634 BufferAssignmentProto buffer_assignment = 3; field
144 ObjectsAssignment<size_t>* buffer_assignment, in GetBufferAsignment()484 ObjectsAssignment<size_t> buffer_assignment; in AllocateBufferBasedTensors() local999 ObjectsAssignment<size_t> buffer_assignment; in GetTotalBufferSizeForTensors() local
26 const BufferAssignment& buffer_assignment) { in CreateBufferInfosFromBufferAssignment()
87 const BufferAssignment& buffer_assignment() const { return *assignment_; } in buffer_assignment() function
603 ObjectsAssignment<size_t> buffer_assignment; in AssignInternalObjects() local
515 ObjectsAssignment<size_t> buffer_assignment; in AllocateMemoryForBuffers() local