Home
last modified time | relevance | path

Searched refs:param_shp_min (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dgather.cc70 ShapeVector param_shp_min = (param_dyn) ? params->shape()->min_shape() : params->shape()->shape(); in GatherInfer() local
91 ShapeVector min_shape = calc_shape(indices_shp_min, param_shp_min); in GatherInfer()
/third_party/mindspore/mindspore/core/abstract/
Dprim_arrays.cc609 ShapeVector param_shp_min = (param_dyn) ? params->shape()->min_shape() : params->shape()->shape(); in InferImplGatherV2() local
630 ShapeVector min_shape = calc_shape(indices_shp_min, param_shp_min); in InferImplGatherV2()