Searched defs:shape_v (Results 1 – 2 of 2) sorted by relevance
43 auto shape_v = GetValue<std::vector<int64_t>>(shape->BuildValue()); in ReshapeInfer() local
338 std::vector<int> shape_v; in BuildAttentionPrim() local