Searched refs:right_dims (Results 1 – 2 of 2) sorted by relevance
51 llvm::ArrayRef<int64_t> right_dims, in TransposeReshape() argument62 for (auto dim : right_dims) { in TransposeReshape()69 transpose_permutation.append(right_dims.begin(), right_dims.end()); in TransposeReshape()
230 const Dimensions& right_dims = m_rightImpl.dimensions();235 right_index += (subs[i] % right_dims[i]) * m_rightStrides[i];240 right_index += (subs[i] % right_dims[i]) * m_rightStrides[i];335 const Dimensions& right_dims = this->m_rightImpl.dimensions();338 right_index += (subs[i] % right_dims[i]) * this->m_rightStrides[i];