Searched defs:size_splits (Results 1 – 5 of 5) sorted by relevance
38 const TfLiteTensor* size_splits; member50 void GetSizeSplitsVector(const TfLiteTensor* size_splits, in GetSizeSplitsVector()60 const TfLiteTensor* size_splits, in ResizeOutputTensors()134 auto size_splits = op_context.size_splits; in Prepare() local
31 SplitVOpModel(const TensorData& input, const TensorData& size_splits, in SplitVOpModel()
57 static Graph* MakeGraph(int split_dim, const std::vector<int64>& size_splits, in MakeGraph()
241 std::vector<Tensor> SplitVGrad(const Tensor& x, const Tensor& size_splits, in SplitVGrad()297 Tensor size_splits(DT_INT64, {2}); in TEST() local
611 const Tensor* size_splits = c->input_tensor(1); in __anon7c94107b0b02() local