Searched refs:calc_shape (Results 1 – 2 of 2) sorted by relevance
82 …auto calc_shape = [axis_val](const ShapeVector &ind_vec, const ShapeVector ¶ms_vec) -> ShapeVe… in GatherInfer() local89 ShapeVector out_shape = calc_shape(indices_shp, params_shp); in GatherInfer()91 ShapeVector min_shape = calc_shape(indices_shp_min, param_shp_min); in GatherInfer()92 ShapeVector max_shape = calc_shape(indices_shp_max, param_shp_max); in GatherInfer()
621 …auto calc_shape = [axis_val](const ShapeVector &ind_vec, const ShapeVector ¶ms_vec) -> ShapeVe… in InferImplGatherV2() local628 ShapeVector out_shape = calc_shape(indices_shp, params_shp); in InferImplGatherV2()630 ShapeVector min_shape = calc_shape(indices_shp_min, param_shp_min); in InferImplGatherV2()631 ShapeVector max_shape = calc_shape(indices_shp_max, param_shp_max); in InferImplGatherV2()