Searched refs:indices_min_shape (Results 1 – 2 of 2) sorted by relevance
68 auto indices_min_shape = indices->shape()->min_shape(); in EmbeddingLookupInfer() local70 if (!indices_min_shape.empty()) { in EmbeddingLookupInfer()71 (void)min_shape.insert(min_shape.end(), indices_min_shape.begin(), indices_min_shape.end()); in EmbeddingLookupInfer()
675 auto indices_min_shape = indices_shp->min_shape(); in InferImplEmbeddingLookup() local687 if (!indices_min_shape.empty()) { in InferImplEmbeddingLookup()688 min_shape.insert(min_shape.end(), indices_min_shape.begin(), indices_min_shape.end()); in InferImplEmbeddingLookup()