tensorflow/examples/android/jni/jni_utils.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

#ifndef ORG_TENSORFLOW_JNI_JNI_UTILS_H_  // NOLINT
#define ORG_TENSORFLOW_JNI_JNI_UTILS_H_  // NOLINT

#include <jni.h>
#include <string>
#include <vector>

#include "tensorflow/core/platform/port.h"

namespace google {
namespace protobuf {
class MessageLite;
}  // google
}  // protobuf

class AAssetManager;

bool PortableReadFileToProto(const std::string& file_name,
                             ::google::protobuf::MessageLite* proto);

void ReadFileToProto(AAssetManager* const asset_manager,
    const char* const filename, google::protobuf::MessageLite* message);

void ReadFileToString(AAssetManager* const asset_manager,
    const char* const filename, std::string* str);

void ReadFileToVector(AAssetManager* const asset_manager,
    const char* const filename, std::vector<std::string>* str_vector);

#endif  // ORG_TENSORFLOW_JNI_JNI_UTILS_H_