Home
last modified time | relevance | path

Searched refs:GetTensorSize (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dshape_test.cc45 int32_t GetOutputSize() { return GetTensorSize(output_); } in GetOutputSize()
Done_hot_test.cc56 int32_t GetOutputSize() { return GetTensorSize(output_); } in GetOutputSize()
Dtest_util.cc149 int32_t SingleOpModel::GetTensorSize(int index) const { in GetTensorSize() function in tflite::SingleOpModel
Dtest_util.h317 return std::vector<T>(v, v + GetTensorSize(index)); in ExtractVector()
338 int32_t GetTensorSize(int index) const;