Searched refs:size_splits_shape (Results 1 – 2 of 2) sorted by relevance
81 std::initializer_list<int> size_splits_shape, in Check() argument88 {TensorType_INT32, size_splits_shape}, num_splits, in Check()100 {TensorType_INT32, size_splits_shape}, num_splits, in Check()
2139 size_splits_shape = size_splits._shape_tuple()2140 if size_splits_shape:2141 num = size_splits_shape[0]