Home
last modified time | relevance | path

Searched refs:UniqueTfLiteTensor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dbuffer_map_test.cc39 using UniqueTfLiteTensor = typedef
43 UniqueTfLiteTensor MakeLiteTensor(const std::vector<int>& shape, in MakeLiteTensor()
45 auto tensor = UniqueTfLiteTensor(new TfLiteTensor(), [](TfLiteTensor* t) { in MakeLiteTensor()
59 UniqueTfLiteTensor MakeLiteTensor<string>(const std::vector<int>& shape, in MakeLiteTensor()
61 auto tensor = UniqueTfLiteTensor(new TfLiteTensor(), [](TfLiteTensor* t) { in MakeLiteTensor()
110 UniqueTfLiteTensor t = in TEST()
128 UniqueTfLiteTensor t = in TEST()
144 UniqueTfLiteTensor t1 = in TEST()
146 UniqueTfLiteTensor t2 = in TEST()
158 UniqueTfLiteTensor t1 = in TEST()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/
H A Dtest_util.cc28 TfLiteTensor* UniqueTfLiteTensor::get() { return tensor_; } in get()
30 TfLiteTensor& UniqueTfLiteTensor::operator*() { return *tensor_; } in operator *()
32 TfLiteTensor* UniqueTfLiteTensor::operator->() { return tensor_; } in operator ->()
34 const TfLiteTensor* UniqueTfLiteTensor::operator->() const { return tensor_; } in operator ->()
36 void UniqueTfLiteTensor::reset(TfLiteTensor* tensor) { tensor_ = tensor; } in reset()
38 UniqueTfLiteTensor::~UniqueTfLiteTensor() { TfLiteTensorFree(tensor_); } in ~UniqueTfLiteTensor()
H A Dtest_util_test.cc31 auto t_int32 = UniqueTfLiteTensor(interpreter.tensor(0)); in TEST()
35 auto t_int64 = UniqueTfLiteTensor(interpreter.tensor(1)); in TEST()
39 auto t_str = UniqueTfLiteTensor(interpreter.tensor(2)); in TEST()
49 auto t_int32 = UniqueTfLiteTensor(interpreter.tensor(0)); in TEST()
54 t_int32 = UniqueTfLiteTensor(interpreter.tensor(0)); in TEST()
H A Dtflite_tensor_view_test.cc40 auto owned_tflite_tensor = UniqueTfLiteTensor(tflite_tensor); in TEST()
62 auto owned_tflite_tensor = UniqueTfLiteTensor(tflite_tensor); in IntTest()
90 auto owned_tflite_tensor = UniqueTfLiteTensor(tflite_tensor); in FloatTest()
113 auto owned_tflite_tensor = UniqueTfLiteTensor(tflite_tensor); in TEST()
156 auto owned_tflite_tensor = UniqueTfLiteTensor(tflite_tensor); in TEST()
H A Dtest_util.h28 class UniqueTfLiteTensor {
30 explicit UniqueTfLiteTensor(TfLiteTensor* tensor) : tensor_(tensor) {} in UniqueTfLiteTensor() function
32 UniqueTfLiteTensor() = default;
52 ~UniqueTfLiteTensor();