Searched refs:param_shp_max (Results 1 – 2 of 2) sorted by relevance
71 ShapeVector param_shp_max = (param_dyn) ? params->shape()->max_shape() : params->shape()->shape(); in GatherInfer() local92 ShapeVector max_shape = calc_shape(indices_shp_max, param_shp_max); in GatherInfer()
610 ShapeVector param_shp_max = (param_dyn) ? params->shape()->max_shape() : params->shape()->shape(); in InferImplGatherV2() local631 ShapeVector max_shape = calc_shape(indices_shp_max, param_shp_max); in InferImplGatherV2()