Searched defs:BufferAssignment (Results 1 – 3 of 3) sorted by relevance
218 struct BufferAssignment { struct220 std::vector<int> buffers_size;223 std::map<int, TupleShape> buffers_shape;226 std::map<int, int> param_to_alloc_idx;229 int output_idx = -1;
290 friend class BufferAssignment; variable500 BufferAssignment(const HloModule* module, in BufferAssignment() function
33 class BufferAssignment; variable