Searched refs:bdims (Results 1 – 2 of 2) sorted by relevance
152 int bdims = b_shape.rank(); in SolveWithInvertedDiagonalBlocks() local153 int64 block_dim = (left_side) ? bdims - 2 : bdims - 1; in SolveWithInvertedDiagonalBlocks()
614 std::vector<int64> bdims(dims.begin() + skip, dims.end()); in BroadcastableDims() local619 for (int64& dim : bdims) { in BroadcastableDims()627 dims.swap(bdims); in BroadcastableDims()629 return {dims, bdims}; in BroadcastableDims()