| /external/tensorflow/tensorflow/lite/kernels/ |
| transpose_conv.cc | 31 namespace tflite { namespace
213 tflite::ConvParams op_params;
269 } // namespace tflite
|
| unidirectional_sequence_rnn.cc | 29 namespace tflite { namespace
334 } // namespace tflite
|
| bidirectional_sequence_rnn_test.cc | 15 // Unit test for TFLite Bidirectional RNN op.
27 namespace tflite { namespace
[all...] |
| fully_connected.cc | 36 namespace tflite { namespace
541 } // namespace tflite
|
| fully_connected_test.cc | 15 // Unit test for TFLite FULLY_CONNECTED op.
30 namespace tflite { namespace
880 } // namespace tflite
883 ::tflite::LogToStderr();
|
| test_util.h | 33 namespace tflite { namespace
341 std::unique_ptr<tflite::Interpreter> interpreter_;
525 } // namespace tflite
|
| unidirectional_sequence_lstm_test.cc | 15 // Unit test for TFLite Sequential LSTM op.
27 namespace tflite { namespace
[all...] |
| /external/tensorflow/tensorflow/lite/ |
| model.cc | 33 namespace tflite { namespace
156 const tflite::Model* caller_owned_model_spec,
167 if (!tflite::ModelBufferHasIdentifier(allocation_->base())) {
171 ident[0], ident[1], ident[2], ident[3], tflite::ModelIdentifier());
187 model_ = ::tflite::GetModel(allocation_->base());
199 InterpreterBuilder::InterpreterBuilder(const ::tflite::Model* model,
360 auto get_name = [](const tflite::Tensor* t) -> const char* {
526 const tflite::SubGraph* subgraph = (*subgraphs)[subgraph_index];
527 tflite::Subgraph* modified_subgraph =
569 } // namespace tflite
[all...] |
| nnapi_delegate.cc | 31 namespace tflite { namespace
45 __android_log_vprint(ANDROID_LOG_ERROR, "tflite", format, args_for_log);
57 FATAL("Aborting since tflite returned failure nnapi_delegate.cc:%d.", \
70 "Returning error since TFLite returned failure nnapi_delegate.cc:%d.", \
115 TfLiteStatus addTensorOperands(tflite::Subgraph* subgraph,
128 // NNAPI requires 32-bit float scale to be zero, tflite doesn't care
215 tflite::Subgraph* subgraph, ANeuralNetworksModel* nn_model,
224 tflite::BuiltinOperator builtin =
225 static_cast<tflite::BuiltinOperator>(registration.builtin_code);
423 case tflite::BuiltinOperator_ADD
[all...] |
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| interpreter_wrapper.cc | 48 namespace tflite { namespace
55 std::unique_ptr<tflite::Interpreter> CreateInterpreter(
56 const tflite::FlatBufferModel* model,
57 const tflite::ops::builtin::BuiltinOpResolver& resolver) {
62 ::tflite::python::ImportNumpy();
64 std::unique_ptr<tflite::Interpreter> interpreter;
65 if (tflite::InterpreterBuilder(*model, resolver)(&interpreter) != kTfLiteOk) {
87 std::unique_ptr<tflite::FlatBufferModel> model,
95 auto resolver = absl::make_unique<tflite::ops::builtin::BuiltinOpResolver>();
109 std::unique_ptr<tflite::FlatBufferModel> model
[all...] |
| /external/tensorflow/tensorflow/lite/testing/ |
| generated_examples_zip_test.cc | 31 namespace tflite { namespace
261 tflite::testing::TfLiteDriver test_driver(FLAGS_use_nnapi);
276 bool result = tflite::testing::ParseAndRunTests(&tflite_stream, &test_driver);
318 } // namespace tflite
321 ::testing::AddGlobalTestEnvironment(tflite::testing::archive_environment());
325 "ignore_known_bugs", &tflite::testing::FLAGS_ignore_known_bugs,
329 "tar_file_path", tflite::testing::FLAGS_tar_file_path,
332 "zip_file_path", tflite::testing::FLAGS_zip_file_path,
335 tflite::testing::FLAGS_unzip_binary_path,
338 tflite::testing::FLAGS_tar_binary_path
[all...] |
| tflite_driver.cc | 28 namespace tflite { namespace
417 } // namespace tflite
|
| /external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| tflite_driver.cc | 30 namespace tflite { namespace
412 } // namespace tflite
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| export_test.cc | 15 #include "tensorflow/lite/toco/tflite/export.h"
20 #include "tensorflow/lite/toco/tflite/builtin_operator.h"
21 #include "tensorflow/lite/toco/tflite/operator.h"
22 #include "tensorflow/lite/toco/tflite/types.h"
26 namespace tflite { namespace in namespace:toco
78 // Even though assert is known to TOCO, it doesn't have a tflite
159 auto* model = ::tflite::GetModel(result.data());
161 for (const ::tflite::OperatorCode* opcode : *model->operator_codes()) {
162 if (opcode->builtin_code() != ::tflite::BuiltinOperator_CUSTOM) {
163 names.push_back(string("builtin:") + ::tflite::EnumNameBuiltinOperator
[all...] |
| import.cc | 15 #include "tensorflow/lite/toco/tflite/import.h"
20 #include "tensorflow/lite/toco/tflite/operator.h"
21 #include "tensorflow/lite/toco/tflite/types.h"
27 namespace tflite { namespace in namespace:toco
30 void LoadTensorsTable(const ::tflite::Model& input_model,
40 void LoadOperatorsTable(const ::tflite::Model& input_model,
45 if (opcode->builtin_code() != ::tflite::BuiltinOperator_CUSTOM) {
55 void ImportTensors(const ::tflite::Model& input_model, Model* model) {
101 const ::tflite::Model& input_model,
169 const ::tflite::Model& input_model
[all...] |
| operator_test.cc | 15 #include "tensorflow/lite/toco/tflite/operator.h"
28 namespace tflite { namespace in namespace:toco
72 ::tflite::CustomOptionsFormat_FLEXBUFFERS));
74 flatbuffers::GetRoot<::tflite::Operator>(builder.GetBufferPointer());
97 ASSERT_EQ(::tflite::BuiltinOptions_NONE, options.type);
[all...] |
| /external/tensorflow/tensorflow/lite/tools/benchmark/ |
| benchmark_tflite_model.cc | 37 void RegisterSelectedOps(::tflite::MutableOpResolver* resolver);
40 namespace tflite { namespace
43 void ProfilingListener::SetInterpreter(tflite::Interpreter* interpreter) {
128 void FillRandomString(tflite::DynamicBuffer* buffer,
342 tflite::DynamicBuffer buffer;
357 model = tflite::FlatBufferModel::BuildFromFile(graph.c_str());
366 tflite::MutableOpResolver resolver;
369 tflite::ops::builtin::BuiltinOpResolver resolver;
373 tflite::InterpreterBuilder(*model, resolver)(&interpreter, num_threads);
443 } // namespace tflite
[all...] |
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| calibrator.cc | 39 namespace tflite { namespace
47 // TfLite kernel invocations are C functions and cannot look at the global
206 if (has_no_buffer && tensor->type() == tflite::TensorType_FLOAT32) {
219 tflite::Interpreter* const interpreter,
238 string GetOpName(const tflite::OperatorCode& opcode) {
242 return tflite::EnumNamesBuiltinOperator()[opcode.builtin_code()];
278 // At runtime TFLite kernel invoke functions can only look into their
319 tflite::InterpreterBuilder(model, *logging_op_resolver)(interpreter);
347 } // namespace tflite
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| quantize_weights.cc | 31 namespace tflite { namespace
262 LOG(ERROR) << "Quantize weights tool only supports tflite models with one "
339 // consumer, since TFLite requires operators to be topo-sorted.
384 } // namespace tflite
|
| subgraph_quantizer.cc | 31 namespace tflite { namespace
385 } // namespace tflite
|
| subgraph_quantizer_test.cc | 30 namespace tflite { namespace
69 tflite::ModelT model;
168 tflite::ModelT model;
280 tflite::ModelT model;
337 tflite::ModelT model;
394 tflite::ModelT model;
449 tflite::ModelT model;
502 } // namespace tflite
508 "Path to test tflite model file."),
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| nnapi_delegate.cc | 32 namespace tflite { namespace
46 __android_log_vprint(ANDROID_LOG_ERROR, "tflite", format, args_for_log);
58 FATAL("Aborting since tflite returned failure nnapi_delegate.cc:%d.", \
71 "Returning error since TFLite returned failure nnapi_delegate.cc:%d.", \
145 TfLiteStatus addTensorOperands(tflite::Interpreter* interpreter,
157 // NNAPI requires 32-bit float scale to be zero, tflite doesn't care
247 tflite::Interpreter* interpreter, ANeuralNetworksModel* nn_model,
255 tflite::BuiltinOperator builtin =
256 static_cast<tflite::BuiltinOperator>(registration.builtin_code);
452 case tflite::BuiltinOperator_ADD
[all...] |
| /external/tensorflow/tensorflow/lite/core/ |
| subgraph.cc | 24 namespace tflite { namespace
675 // automatic tflite CPU fallback.
[all...] |
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| reference_ops.h | 37 namespace tflite { namespace
44 tflite::ArithmeticParams* params) {
290 inline void DepthToSpace(const tflite::DepthToSpaceParams& op_params,
343 inline void SpaceToDepth(const tflite::SpaceToDepthParams& op_params,
441 inline void ReluX(const tflite::ActivationParams& params,
456 inline void LeakyRelu(const tflite::LeakyReluParams& params,
469 inline void L2Normalization(const tflite::L2NormalizationParams& op_params,
492 inline void L2Normalization(const tflite::L2NormalizationParams& op_params,
[all...] |
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| tensor_utils_test.cc | 24 namespace tflite { namespace
[all...] |
