Searched defs:BufferAssignment (Results 1 – 3 of 3) sorted by relevance
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;
32 class BufferAssignment; variable
292 friend class BufferAssignment; variable499 BufferAssignment(const HloModule* module, in BufferAssignment() function