Searched defs:BufferAssignment (Results 1 – 3 of 3) sorted by relevance
32 class BufferAssignment; variable
278 friend class BufferAssignment; variable483 BufferAssignment(const HloModule* module, in BufferAssignment() function
197 struct BufferAssignment { struct199 std::vector<int> buffers_size;202 std::map<int, TupleShape> buffers_shape;205 std::vector<int> params_idx;208 int output_idx = -1;