/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
|