Searched refs:params_shp (Results 1 – 6 of 6) sorted by relevance
54 auto params_shp = params->shape()->shape(); in EmbeddingLookupInfer() local59 (void)shape.insert(shape.end(), params_shp.begin() + 1, params_shp.end()); in EmbeddingLookupInfer()64 (void)max_shape.insert(max_shape.end(), params_shp.begin() + 1, params_shp.end()); in EmbeddingLookupInfer()72 (void)min_shape.insert(min_shape.end(), params_shp.begin() + 1, params_shp.end()); in EmbeddingLookupInfer()
65 auto params_shp = params->shape()->shape(); in GatherInfer() local67 auto params_rank = static_cast<int64_t>(params_shp.size()); in GatherInfer()89 ShapeVector out_shape = calc_shape(indices_shp, params_shp); in GatherInfer()
117 params_shp = params['shape']118 rank = len(params_shp)122 out_shape = params_shp[:axis_v] + indices['shape'] + params_shp[axis_v + 1:]
605 auto params_shp = params->shape()->shape(); in InferImplGatherV2() local607 auto params_rank = static_cast<int64_t>(params_shp.size()); in InferImplGatherV2()628 ShapeVector out_shape = calc_shape(indices_shp, params_shp); in InferImplGatherV2()667 auto params_shp = params->shape(); in InferImplEmbeddingLookup() local668 MS_EXCEPTION_IF_NULL(params_shp); in InferImplEmbeddingLookup()669 auto params_shape = params_shp->shape(); in InferImplEmbeddingLookup()
391 auto params_shp = params_ori_shp->cast<abstract::ShapePtr>(); in CreateUpdateCache() local392 MS_EXCEPTION_IF_NULL(params_shp); in CreateUpdateCache()393 auto params_shape = params_shp->shape(); in CreateUpdateCache()
5883 params_shp = params['shape']5884 if len(params_shp) > 2: