Searched refs:indices_max_shape (Results 1 – 2 of 2) sorted by relevance
60 auto indices_max_shape = indices->shape()->max_shape(); in EmbeddingLookupInfer() local62 if (!indices_max_shape.empty()) { in EmbeddingLookupInfer()63 (void)max_shape.insert(max_shape.end(), indices_max_shape.begin(), indices_max_shape.end()); in EmbeddingLookupInfer()
674 auto indices_max_shape = indices_shp->max_shape(); in InferImplEmbeddingLookup() local681 if (!indices_max_shape.empty()) { in InferImplEmbeddingLookup()682 max_shape.insert(max_shape.end(), indices_max_shape.begin(), indices_max_shape.end()); in InferImplEmbeddingLookup()