Searched defs:all_shape (Results 1 – 2 of 2) sorted by relevance
64 …auto all_shape = _get_pack_shape(x_shapes, x_types, GetValue<int64_t>(primitive->GetAttr(kAxis)), … in PackInfer() local
475 std::vector<float> all_shape; in InitPSOptimInputShapes() local