Searched refs:last_blocks_dims (Results 1 – 1 of 1) sorted by relevance
116 auto last_blocks_dims = std::vector<int64>(ndims); in DiagonalBlocks() local117 std::copy(shape_dims.begin(), shape_dims.end(), last_blocks_dims.begin()); in DiagonalBlocks()118 last_blocks_dims.insert(last_blocks_dims.end() - 2, 1); in DiagonalBlocks()119 last_blocks = Reshape(last_blocks, last_blocks_dims); in DiagonalBlocks()