Home
last modified time | relevance | path

Searched refs:typed_tensor (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/examples/label_image/
Dbitmap_helpers_impl.h71 auto input = interpreter->typed_tensor<float>(0); in resize()
77 interpreter->typed_tensor<int>(1)[0] = wanted_height; in resize()
78 interpreter->typed_tensor<int>(1)[1] = wanted_width; in resize()
82 auto output = interpreter->typed_tensor<float>(2); in resize()
Dlabel_image.cc172 resize<float>(interpreter->typed_tensor<float>(input), in.data(), in RunInference()
177 resize<uint8_t>(interpreter->typed_tensor<uint8_t>(input), in.data(), in RunInference()
/external/tensorflow/tensorflow/lite/
Dinterpreter.h270 T* typed_tensor(int tensor_index) { in typed_tensor() function
282 const T* typed_tensor(int tensor_index) const { in typed_tensor() function
295 return typed_tensor<T>(inputs()[index]); in typed_input_tensor()
302 return typed_tensor<T>(inputs()[index]); in typed_input_tensor()
309 return typed_tensor<T>(outputs()[index]); in typed_output_tensor()
316 return typed_tensor<T>(outputs()[index]); in typed_output_tensor()
Dinterpreter_test.cc353 ASSERT_NE(interpreter.typed_tensor<float>(0), nullptr); in TEST()
355 ASSERT_EQ(interpreter.typed_tensor<int>(0), nullptr); in TEST()
357 ASSERT_EQ(interpreter.typed_tensor<float>(0), interpreter.tensor(0)->data.f); in TEST()
797 interpreter.typed_tensor<int>(1)[0] = 2; in TEST()
798 interpreter.typed_tensor<int>(1)[1] = 2; in TEST()
799 interpreter.typed_tensor<int>(1)[2] = 2; in TEST()
800 interpreter.typed_tensor<int>(1)[3] = 2; in TEST()
801 interpreter.typed_tensor<int>(1)[4] = 0; in TEST()
802 interpreter.typed_tensor<int>(1)[5] = 0; in TEST()
803 interpreter.typed_tensor<int>(1)[6] = 0; in TEST()
[all …]
Dnnapi_delegate.cc833 subgraph->typed_tensor<float>(subgraph->outputs()[0])) { in Invoke()
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dparse_testdata.cc167 if (float* data = interpreter->typed_tensor<float>(input_index)) { in FeedExample()
172 interpreter->typed_tensor<int32_t>(input_index)) { in FeedExample()
177 interpreter->typed_tensor<int64_t>(input_index)) { in FeedExample()
200 if (const float* data = interpreter->typed_tensor<float>(output_index)) { in CheckOutputs()
218 interpreter->typed_tensor<int32_t>(output_index)) { in CheckOutputs()
229 interpreter->typed_tensor<int64_t>(output_index)) { in CheckOutputs()
/external/tensorflow/tensorflow/lite/testing/
Dparse_testdata.cc164 if (float* data = interpreter->typed_tensor<float>(input_index)) { in FeedExample()
169 interpreter->typed_tensor<int32_t>(input_index)) { in FeedExample()
174 interpreter->typed_tensor<int64_t>(input_index)) { in FeedExample()
197 if (const float* data = interpreter->typed_tensor<float>(output_index)) { in CheckOutputs()
215 interpreter->typed_tensor<int32_t>(output_index)) { in CheckOutputs()
226 interpreter->typed_tensor<int64_t>(output_index)) { in CheckOutputs()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dtest_util.h50 memcpy(interpreter_->typed_tensor<T>(tensor_index), values.data(), in SetTypedValues()
58 const T* tdata = interpreter_->typed_tensor<T>(tensor_index); in GetTypedValues()
/external/tensorflow/tensorflow/lite/tools/benchmark/
Dbenchmark_tflite_model.cc330 std::memcpy(interpreter->typed_tensor<float>(i), inputs_data_[j].data.f, in ResetInputsAndOutputs()
333 std::memcpy(interpreter->typed_tensor<int32_t>(i), in ResetInputsAndOutputs()
336 std::memcpy(interpreter->typed_tensor<uint8_t>(i), in ResetInputsAndOutputs()
339 std::memcpy(interpreter->typed_tensor<int8_t>(i), in ResetInputsAndOutputs()
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.h270 T* v = interpreter_->typed_tensor<T>(index); in PopulateTensor()
290 T* v = interpreter_->typed_tensor<T>(index); in PopulateTensor()
308 T* v = interpreter_->typed_tensor<T>(index); in PopulateTensor()
315 T* v = interpreter_->typed_tensor<T>(index); in ExtractVector()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dios.md142 uint8_t* buffer = interpreter->typed_tensor<uint8_t>(input_tensor_index);
203 uint8_t* buffer = interpreter->typed_tensor<uint8_t>(output_tensor_index);
/external/tensorflow/tensorflow/lite/g3doc/models/image_classification/
Dios.md142 uint8_t* buffer = interpreter->typed_tensor<uint8_t>(input_tensor_index);
203 uint8_t* buffer = interpreter->typed_tensor<uint8_t>(output_tensor_index);
/external/tensorflow/tensorflow/lite/examples/ios/simple/
DRunModelViewController.mm164 float* out = interpreter->typed_tensor<float>(input);
/external/tensorflow/tensorflow/lite/examples/ios/camera/
DCameraExampleViewController.mm325 uint8_t* out = interpreter->typed_tensor<uint8_t>(input);
328 float* out = interpreter->typed_tensor<float>(input);
/external/tensorflow/tensorflow/contrib/lite/
Dnnapi_delegate.cc845 interpreter->typed_tensor<float>(interpreter->outputs()[0])) { in Invoke()