| /external/tensorflow/tensorflow/contrib/lite/toco/runtime/ |
| types.h | 25 // the code over to tflite.
26 using tflite::Dims;
27 using tflite::FusedActivationFunctionType;
28 using tflite::RequiredBufferSizeForDims;
|
| /external/tensorflow/tensorflow/contrib/lite/java/src/main/native/ |
| builtin_ops_jni.cc | 18 namespace tflite { namespace
21 // the tflite namespace. This one instantiates a BuiltinOpResolver, with all the
25 return std::unique_ptr<tflite::ops::builtin::BuiltinOpResolver>(
26 new tflite::ops::builtin::BuiltinOpResolver());
29 } // namespace tflite
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| string_tflite.h | 21 namespace tflite { namespace
27 } // namespace tflite
|
| optional_debug_tools.h | 22 namespace tflite { namespace
30 } // namespace tflite
|
| memory_planner.h | 20 namespace tflite { namespace
43 } // namespace tflite
|
| nnapi_delegate.cc | 26 namespace tflite { namespace
41 FATAL("Aborting since tflite returned failure."); \
46 FATAL("Aborting since tflite returned failure."); \
74 uint32_t addTensorOperands(tflite::Interpreter* interpreter,
137 void AddOpsAndParams(tflite::Interpreter* interpreter,
143 tflite::BuiltinOperator builtin =
144 static_cast<tflite::BuiltinOperator>(registration.builtin_code);
256 case tflite::BuiltinOperator_ADD:
260 case tflite::BuiltinOperator_AVERAGE_POOL_2D:
264 case tflite::BuiltinOperator_MAX_POOL_2D
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/testing/ |
| util.h | 18 namespace tflite { namespace
26 } // namespace tflite
|
| parse_testdata.h | 22 namespace tflite { namespace
43 TfLiteStatus FeedExample(tflite::Interpreter* interpreter, const Example&);
46 TfLiteStatus CheckOutputs(tflite::Interpreter* interpreter, const Example&);
72 } // namespace tflite
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| types.cc | 15 #include "tensorflow/contrib/lite/toco/tflite/types.h"
19 namespace tflite { namespace in namespace:toco
33 void CopyBuffer(const ::tflite::Buffer& buffer, Array* array) {
48 ::tflite::TensorType DataType::Serialize(ArrayDataType array_data_type) {
51 return ::tflite::TensorType_FLOAT32;
53 return ::tflite::TensorType_INT32;
55 return ::tflite::TensorType_INT64;
57 return ::tflite::TensorType_UINT8;
59 return ::tflite::TensorType_STRING;
63 return ::tflite::TensorType_FLOAT32
[all...] |
| types.h | 23 namespace tflite { namespace in namespace:toco
26 static ::tflite::TensorType Serialize(ArrayDataType array_data_type);
39 static void Deserialize(const ::tflite::Tensor& tensor,
40 const ::tflite::Buffer& buffer, Array* array);
44 static ::tflite::Padding Serialize(PaddingType padding_type);
49 static ::tflite::ActivationFunctionType Serialize(
54 } // namespace tflite
|
| import.h | 23 namespace tflite { namespace in namespace:toco
39 void LoadTensorsTable(const ::tflite::Model &input_model,
41 void LoadOperatorsTable(const ::tflite::Model &input_model,
45 } // namespace tflite
|
| import_test.cc | 15 #include "tensorflow/contrib/lite/toco/tflite/import.h"
25 namespace tflite { namespace in namespace:toco
44 auto q = ::tflite::CreateQuantizationParameters(
50 auto buf0 = ::tflite::CreateBuffer(builder_, CreateDataVector<float>({}));
52 ::tflite::CreateBuffer(builder_, CreateDataVector<float>({1.0f, 2.0f}));
54 ::tflite::CreateBuffer(builder_, CreateDataVector<float>({3.0f}));
56 std::vector<flatbuffers::Offset<::tflite::Buffer>>({buf0, buf1, buf2}));
57 auto t1 = ::tflite::CreateTensor(builder_,
59 ::tflite::TensorType_FLOAT32, 1,
62 ::tflite::CreateTensor(builder_, builder_.CreateVector<int>({2, 1})
[all...] |
| types_test.cc | 15 #include "tensorflow/contrib/lite/toco/tflite/types.h"
22 namespace tflite { namespace in namespace:toco
35 static const ::tflite::TensorType kUnsupportedTfLiteTypes[] = {
36 ::tflite::TensorType_FLOAT16};
59 buffer_builder.Finish(::tflite::CreateBuffer(buffer_builder, data_buffer));
62 flatbuffers::GetRoot<::tflite::Tensor>(builder.GetBufferPointer());
64 flatbuffers::GetRoot<::tflite::Buffer>(buffer_builder.GetBufferPointer());
70 std::vector<std::pair<ArrayDataType, ::tflite::TensorType>> testdata = {
71 {ArrayDataType::kUint8, ::tflite::TensorType_UINT8},
72 {ArrayDataType::kInt32, ::tflite::TensorType_INT32}
[all...] |
| operator.cc | 15 #include "tensorflow/contrib/lite/toco/tflite/operator.h"
17 #include "tensorflow/contrib/lite/toco/tflite/builtin_operator.h"
18 #include "tensorflow/contrib/lite/toco/tflite/custom_operator.h"
19 #include "tensorflow/contrib/lite/toco/tflite/simple_operator.h"
20 #include "tensorflow/contrib/lite/toco/tflite/types.h"
27 namespace tflite { namespace in namespace:toco
30 : public BuiltinOperator<AveragePoolOperator, ::tflite::Pool2DOptions,
31 ::tflite::BuiltinOptions_Pool2DOptions> {
41 return ::tflite::CreatePool2DOptions(*builder, padding, op.stride_width,
59 : public BuiltinOperator<ConvOperator, ::tflite::Conv2DOptions
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/tools/ |
| gen_op_registration.h | 21 namespace tflite { namespace
30 // Read ops from the TFLite model.
33 void ReadOpsFromModel(const ::tflite::Model* model,
37 } // namespace tflite
|
| mutable_op_resolver.h | 25 struct hash<tflite::BuiltinOperator> {
26 size_t operator()(const tflite::BuiltinOperator& op) const {
32 namespace tflite { namespace
43 TfLiteRegistration* FindOp(tflite::BuiltinOperator op) const override;
45 void AddBuiltin(tflite::BuiltinOperator op, TfLiteRegistration* registration);
53 } // namespace tflite
|
| verifier.h | 24 namespace tflite { namespace
34 } // namespace tflite
|
| benchmark_model.cc | 30 void RegisterSelectedOps(::tflite::MutableOpResolver* resolver);
44 std::unique_ptr<tflite::FlatBufferModel> model;
45 std::unique_ptr<tflite::Interpreter> interpreter;
51 model = tflite::FlatBufferModel::BuildFromFile(graph.c_str());
60 tflite::MutableOpResolver resolver;
63 tflite::ops::builtin::BuiltinOpResolver resolver;
66 tflite::InterpreterBuilder(*model, resolver)(&interpreter);
|
| /external/tensorflow/tensorflow/contrib/lite/schema/builtin_ops_header/ |
| generate.cc | 21 if (!tflite::builtin_ops_header::GenerateHeader(std::cout)) {
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| register.h | 22 namespace tflite { namespace
29 TfLiteRegistration* FindOp(tflite::BuiltinOperator op) const override;
31 void AddBuiltin(tflite::BuiltinOperator op, TfLiteRegistration* registration);
36 size_t operator()(const tflite::BuiltinOperator& x) const {
40 std::unordered_map<tflite::BuiltinOperator, TfLiteRegistration*,
48 } // namespace tflite
|
| gemm_support.h | 21 namespace tflite { namespace
52 } // namespace tflite
|
| /external/libtextclassifier/ |
| model-executor.h | 34 bool FromModelSpec(const tflite::Model* model_spec,
35 std::unique_ptr<const tflite::FlatBufferModel>* model);
43 tflite::Interpreter* interpreter);
50 const tflite::Model* model =
51 flatbuffers::GetRoot<tflite::Model>(model_spec_buffer->data());
61 const tflite::Model* model_spec) {
62 std::unique_ptr<const tflite::FlatBufferModel> model;
71 std::unique_ptr<tflite::Interpreter> CreateInterpreter() const;
74 tflite::Interpreter* interpreter) const {
80 explicit ModelExecutor(std::unique_ptr<const tflite::FlatBufferModel> model
[all...] |
| /frameworks/ml/nn/common/operations/ |
| Concatenation.cpp | 29 std::vector<tflite::Dims<4>*> inputDimsPtr(num_inputs);
30 std::vector<tflite::Dims<4> > inputDims(num_inputs);
36 tflite::optimized_ops::Concatenation<tflite::FusedActivationFunctionType::kNone, float>(
48 std::vector<tflite::Dims<4>*> inputDimsPtr(num_inputs);
49 std::vector<tflite::Dims<4> > inputDims(num_inputs);
55 tflite::optimized_ops::Concatenation<tflite::FusedActivationFunctionType::kNone, uint8_t>(
|
| /frameworks/ml/nn/runtime/test/benchmark/libnnbenchmark/ |
| run_tflite.h | 37 std::unique_ptr<tflite::FlatBufferModel> mTfliteModel;
38 std::unique_ptr<tflite::Interpreter> mTfliteInterpreter;
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| kernel_utils.h | 20 namespace tflite { namespace
39 } // namespace tflite
|
