Searched refs:size_splits_ (Results 1 – 1 of 1) sorted by relevance
34 size_splits_ = AddInput(size_splits); in SplitVOpModel()47 {GetShape(input_), GetShape(size_splits_), GetShape(axis_)}); in SplitVOpModel()49 BuildInterpreter({GetShape(input_), GetShape(size_splits_), {}}); in SplitVOpModel()58 PopulateTensor(size_splits_, data); in SetSizeSplits()70 int size_splits_; member in tflite::__anon596d55d20111::SplitVOpModel