Searched refs:size_splits_ (Results 1 – 1 of 1) sorted by relevance
38 size_splits_ = AddInput(size_splits); in SplitVOpModel()51 {GetShape(input_), GetShape(size_splits_), GetShape(axis_)}); in SplitVOpModel()53 BuildInterpreter({GetShape(input_), GetShape(size_splits_), {}}); in SplitVOpModel()62 PopulateTensor(size_splits_, data); in SetSizeSplits()74 int size_splits_; member in tflite::__anon3edbe8b00111::SplitVOpModel