| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| types_test.cc | 15 #include "tensorflow/lite/toco/tflite/types.h"
24 namespace tflite { namespace in namespace:toco
36 static const ::tflite::TensorType kUnsupportedTfLiteTypes[] = {
37 ::tflite::TensorType_FLOAT16};
60 buffer_builder.Finish(::tflite::CreateBuffer(buffer_builder, data_buffer));
63 flatbuffers::GetRoot<::tflite::Tensor>(builder.GetBufferPointer());
65 flatbuffers::GetRoot<::tflite::Buffer>(buffer_builder.GetBufferPointer());
71 std::vector<std::pair<ArrayDataType, ::tflite::TensorType>> testdata = {
72 {ArrayDataType::kUint8, ::tflite::TensorType_UINT8},
73 {ArrayDataType::kInt32, ::tflite::TensorType_INT32}
[all...] |
| /external/tensorflow/tensorflow/lite/tools/benchmark/ |
| benchmark_test.cc | 30 namespace tflite { namespace
56 const tflite::Interpreter* GetInterpreter() { return interpreter.get(); }
104 } // namespace tflite
108 std::vector<tflite::Flag> flags = {
109 tflite::Flag::CreateFlag("graph", &model_path, "Path to model file.")};
112 tflite::Flags::Parse(&argc, const_cast<const char**>(argv), flags);
114 std::cerr << tflite::Flags::Usage(argv[0], flags);
118 ::tflite::LogToStderr();
|
| command_line_flags.h | 23 namespace tflite { namespace
29 // necessary for tflite benchmarking code.
121 } // namespace tflite
|
| logging.h | 25 namespace tflite { namespace
61 } // namespace tflite
64 tflite::logging::LoggingWrapper( \
65 tflite::logging::LoggingWrapper::LogSeverity::severity) \
69 tflite::logging::LoggingWrapper( \
70 tflite::logging::LoggingWrapper::LogSeverity::FATAL, \
|
| /external/tensorflow/tensorflow/lite/tools/evaluation/ |
| evaluation_stage_test.cc | 25 namespace tflite { namespace
175 } // namespace tflite
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| calibration_common.h | 23 namespace tflite { namespace
43 // Information about an operator in the TfLite graph.
60 } // namespace tflite
|
| /external/tensorflow/tensorflow/lite/java/ovic/src/main/java/org/tensorflow/ovic/ |
| OvicClassifier.java | 34 /** Class for running ImageNet classification with a TfLite model. */
44 private Interpreter tflite; field in class:OvicClassifier
77 tflite = new Interpreter(model, new Interpreter.Options().setNumThreads(1));
78 inputDims = TestHelper.getInputDims(tflite, 0);
95 String outputDataType = TestHelper.getOutputDataType(tflite, 0);
110 if (tflite == null) {
117 tflite.run(imgData, labelProbArray);
119 tflite.run(imgData, inferenceOutputArray);
150 if (tflite == null) {
153 Long latency = tflite.getLastNativeInferenceDurationNanoseconds()
[all...] |
| /external/libtextclassifier/utils/tflite/ |
| dist_diversification.cc | 17 #include "utils/tflite/dist_diversification.h"
94 if (tflite::IsConstantTensor(&num_results)) {
100 tflite::SetTensorToDynamic(&output_indices);
113 if (tflite::IsDynamicTensor(&output_indices)) {
145 namespace tflite { namespace
155 } // namespace tflite
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| common.h | 53 namespace tflite { namespace
269 } // namespace tflite
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| flatbuffer_conversions_test.cc | 25 namespace tflite { namespace
144 } // namespace tflite
|
| op_resolver_test.cc | 22 namespace tflite { namespace
192 } // namespace tflite
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| buffer_map.cc | 24 namespace tflite { namespace
62 // representation in TFLITE and TF, so we just need use memcpy().
91 // A string buffer. TFLITE string tensor format is different than
177 } // namespace tflite
|
| delegate.cc | 28 namespace tflite { namespace
65 // Request TFLite to partition the graph and make kernels for each independent
130 tflite::FlexDelegate::Create().release(), [](TfLiteDelegate* delegate) {
131 delete reinterpret_cast<tflite::FlexDelegate*>(delegate);
148 } // namespace tflite
|
| kernel_test.cc | 22 namespace tflite { namespace
418 } // namespace tflite
421 ::tflite::LogToStderr();
|
| test_util.cc | 22 namespace tflite { namespace
195 } // namespace tflite
|
| test_util.h | 22 namespace tflite { namespace
37 // This class creates models with TF and TFLite ops. In order to use this class
92 // Adds a TFLite Mul op. `inputs` contains the indices of the input tensors
122 } // namespace tflite
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| ctc_decoder.h | 26 namespace tflite { namespace
112 } // namespace tflite
|
| /external/tensorflow/tensorflow/lite/experimental/micro/ |
| simple_tensor_allocator_test.cc | 20 namespace tflite { namespace
94 } // namespace tflite
101 tflite::SimpleTensorAllocator allocator(arena, arena_size);
103 const tflite::Tensor* tensor = tflite::Create1dTensor(100);
104 const flatbuffers::Vector<flatbuffers::Offset<tflite::Buffer>>* buffers =
105 tflite::CreateBuffers();
122 tflite::SimpleTensorAllocator allocator(arena, arena_size);
124 const tflite::Tensor* tensor = tflite::Create1dTensor(2000)
[all...] |
| /external/tensorflow/tensorflow/lite/experimental/micro/testing/ |
| test_utils.h | 27 namespace tflite { namespace
180 } // namespace tflite
|
| /external/tensorflow/tensorflow/lite/ |
| graph_info_test.cc | 22 namespace tflite { namespace
281 } // namespace tflite
284 ::tflite::LogToStderr();
|
| /external/tensorflow/tensorflow/lite/java/demo/app/src/main/java/com/example/android/tflitecamerademo/ |
| ImageClassifierFloatInception.java | 53 // https://storage.googleapis.com/download.tensorflow.org/models/tflite/inception_v3_slim_2016_android_2017_11_10.zip
54 return "inceptionv3_slim_2016.tflite";
103 tflite.run(imgData, labelProbArray);
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| activations_test.cc | 22 namespace tflite { namespace
[all...] |
| arg_min_max.cc | 23 namespace tflite { namespace
235 } // namespace tflite
|
| comparisons.cc | 22 namespace tflite { namespace
370 } // namespace tflite
|
| concatenation.cc | 30 namespace tflite { namespace
118 tflite::ConcatenationParams op_params; \
129 tflite::ConcatenationParams op_params; \
216 } // namespace tflite
|
