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