Searched refs:size_splits_shape (Results 1 – 2 of 2) sorted by relevance
77 std::initializer_list<int> size_splits_shape, in Check() argument83 SplitVOpModel m({T1, input_shape}, {TensorType_INT32, size_splits_shape}, in Check()95 {TensorType_INT32, size_splits_shape}, num_splits, in Check()
1532 size_splits_shape = size_splits._shape_tuple()1533 if size_splits_shape:1534 num = size_splits_shape[0]