Searched refs:normalized_shape_index (Results 1 – 1 of 1) sorted by relevance
4114 const IrArray::Index& normalized_shape_index, in GetUnnormalizedIndex() argument4117 DCHECK_EQ(normalized_shape_index.size(), 3); 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()4126 CHECK_EQ(normalized_shape_index.dims()[0], 1); in GetUnnormalizedIndex()4127 auto multidim = normalized_shape_index.multidim(); in GetUnnormalizedIndex()4129 normalized_shape_index.GetType()); in GetUnnormalizedIndex()4131 llvm::Value* linear = normalized_shape_index.Linearize( in GetUnnormalizedIndex()