Searched refs:kSizeSplits (Results 1 – 3 of 3) sorted by relevance
30 (void)this->AddAttr(kSizeSplits, MakeValue(size_splits)); in set_size_splits()36 auto value_ptr = GetAttr(kSizeSplits); in get_size_splits()
42 auto size_splits = GetValue<std::vector<int64_t>>(primitive->GetAttr(kSizeSplits)); in InferShape()
220 constexpr auto kSizeSplits = "size_splits"; variable