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