Searched defs:BufferAssignment (Results 1 – 3 of 3) sorted by relevance
32 class BufferAssignment; variable
289 friend class BufferAssignment; variable497 BufferAssignment(const HloModule* module, in BufferAssignment() function
199 struct BufferAssignment { struct201 std::vector<int> buffers_size;204 std::map<int, TupleShape> buffers_shape;207 std::map<int, int> param_to_alloc_idx;210 int output_idx = -1;