MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetSizeSplitsVector

Function GetSizeSplitsVector

tensorflow/lite/kernels/split_v.cc:50–56  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

48
49template <typename T>
50void GetSizeSplitsVector(const TfLiteTensor* size_splits,
51 std::vector<int64_t>* size_splits_vector) {
52 const auto num_elements = NumElements(size_splits);
53 for (int i = 0; i < num_elements; ++i) {
54 size_splits_vector->push_back(GetTensorData<T>(size_splits)[i]);
55 }
56}
57
58TfLiteStatus ResizeOutputTensors(TfLiteContext* context, TfLiteNode* node,
59 const TfLiteTensor* input,

Callers

nothing calls this directly

Calls 2

NumElementsFunction · 0.70
push_backMethod · 0.45

Tested by

no test coverage detected