Searched refs:indices_shp_max (Results 1 – 2 of 2) sorted by relevance
73 …ShapeVector indices_shp_max = (ind_dyn) ? indices->shape()->max_shape() : indices->shape()->shape(… in GatherInfer() local92 ShapeVector max_shape = calc_shape(indices_shp_max, param_shp_max); in GatherInfer()
612 …ShapeVector indices_shp_max = (ind_dyn) ? indices->shape()->max_shape() : indices->shape()->shape(… in InferImplGatherV2() local631 ShapeVector max_shape = calc_shape(indices_shp_max, param_shp_max); in InferImplGatherV2()