Searched refs:indices_shp (Results 1 – 6 of 6) sorted by relevance
216 auto indices_shp = indices->shape()->shape(); in InferImplMakeRowTensor() local217 if (indices_shp.size() != 1) { in InferImplMakeRowTensor()218 …MS_EXCEPTION(TypeError) << "Indices must be a 1 dimension tensor, but got a " << indices_shp.size() in InferImplMakeRowTensor()222 if (indices_shp[0] != values_shp[0]) { in InferImplMakeRowTensor()224 << values_shp[0] << ", but got " << indices_shp[0]; in InferImplMakeRowTensor()322 auto indices_shp = indices->shape()->shape(); in InferImplMakeSparseTensor() local323 if (indices_shp.size() != 2) { in InferImplMakeSparseTensor()324 …MS_EXCEPTION(TypeError) << "Indices must be a 2 dimension tensor, but got a " << indices_shp.size() in InferImplMakeSparseTensor()332 if (indices_shp[0] != values_shp[0]) { in InferImplMakeSparseTensor()334 << values_shp[0] << ", but got " << indices_shp[0]; in InferImplMakeSparseTensor()[all …]
438 auto indices_shp = indices->shape(); in InferImplMapCacheIdx() local439 MS_EXCEPTION_IF_NULL(indices_shp); in InferImplMapCacheIdx()444 if (!indices_shp->max_shape().empty()) { in InferImplMapCacheIdx()445 max_shape = indices_shp->max_shape(); in InferImplMapCacheIdx()447 max_shape = indices_shp->shape(); in InferImplMapCacheIdx()606 auto indices_shp = indices->shape()->shape(); in InferImplGatherV2() local628 ShapeVector out_shape = calc_shape(indices_shp, params_shp); in InferImplGatherV2()671 auto indices_shp = indices->shape(); in InferImplEmbeddingLookup() local672 MS_EXCEPTION_IF_NULL(indices_shp); in InferImplEmbeddingLookup()673 auto indices_shape = indices_shp->shape(); in InferImplEmbeddingLookup()[all …]
56 auto indices_shp = indices->shape()->shape(); in EmbeddingLookupInfer() local58 (void)shape.insert(shape.end(), indices_shp.begin(), indices_shp.end()); in EmbeddingLookupInfer()
66 auto indices_shp = indices->shape()->shape(); in GatherInfer() local89 ShapeVector out_shape = calc_shape(indices_shp, params_shp); in GatherInfer()
312 auto indices_shp = indices_ori_shp->cast<abstract::ShapePtr>(); in CreateMapCacheIdx() local316 if (!indices_shp->max_shape().empty()) { in CreateMapCacheIdx()317 max_shape = indices_shp->max_shape(); in CreateMapCacheIdx()319 max_shape = indices_shp->shape(); in CreateMapCacheIdx()326 auto cache_idx = std::make_shared<abstract::AbstractTensor>(indices_element_type, indices_shp); in CreateMapCacheIdx()
5880 indices_shp = indices['shape']5881 if not indices_shp: