Searched refs:left_dims (Results 1 – 2 of 2) sorted by relevance
213 const Dimensions& left_dims = m_leftImpl.dimensions();214 if (subs[m_axis] < left_dims[m_axis]) {219 left_index += (subs[i] % left_dims[i]) * m_leftStrides[i];224 left_index += (subs[i] % left_dims[i]) * m_leftStrides[i];229 subs[m_axis] -= left_dims[m_axis];326 const Dimensions& left_dims = this->m_leftImpl.dimensions();327 if (subs[this->m_axis] < left_dims[this->m_axis]) {330 left_index += (subs[i] % left_dims[i]) * this->m_leftStrides[i];334 subs[this->m_axis] -= left_dims[this->m_axis];
50 llvm::ArrayRef<int64_t> left_dims, in TransposeReshape() argument57 for (auto dim : left_dims) { in TransposeReshape()67 llvm::SmallVector<int64_t, 5> transpose_permutation(left_dims.begin(), in TransposeReshape()68 left_dims.end()); in TransposeReshape()