Searched refs:unnormalized_shape (Results 1 – 1 of 1) sorted by relevance
4115 const Shape& unnormalized_shape, llvm::IRBuilder<>* b_, in GetUnnormalizedIndex() argument4122 if (unnormalized_shape.rank() == 2 && unnormalized_shape.has_layout() && in GetUnnormalizedIndex()4123 unnormalized_shape.dimensions()[0] == normalized_shape_index.dims()[1] && in GetUnnormalizedIndex()4124 unnormalized_shape.dimensions()[1] == normalized_shape_index.dims()[2] && in GetUnnormalizedIndex()4125 unnormalized_shape.layout().minor_to_major(1) == 0) { in GetUnnormalizedIndex()4128 return IrArray::Index({multidim[1], multidim[2]}, unnormalized_shape, in GetUnnormalizedIndex()4133 return IrArray::Index(linear, unnormalized_shape, b_); in GetUnnormalizedIndex()