Add dependecies
This commit is contained in:
parent
68e3ce2fe7
commit
6ffd6820ff
@ -18,6 +18,7 @@ limitations under the License.
|
|||||||
#include "tensorflow/c/c_api_internal.h"
|
#include "tensorflow/c/c_api_internal.h"
|
||||||
#include "tensorflow/c/tf_status_helper.h"
|
#include "tensorflow/c/tf_status_helper.h"
|
||||||
#include "tensorflow/core/platform/env.h"
|
#include "tensorflow/core/platform/env.h"
|
||||||
|
#include "tensorflow/core/platform/path.h"
|
||||||
#include "tensorflow/core/platform/types.h"
|
#include "tensorflow/core/platform/types.h"
|
||||||
|
|
||||||
struct TF_StringStream {
|
struct TF_StringStream {
|
||||||
@ -148,9 +149,9 @@ TF_StringStream* TF_GetLocalTempDirectories() {
|
|||||||
|
|
||||||
void TF_GetTempFileName(const char* extension, std::string* name,
|
void TF_GetTempFileName(const char* extension, std::string* name,
|
||||||
TF_Status* status) {
|
TF_Status* status) {
|
||||||
*name = ::tensorflow::Env::Default()->GetTempFilename(extension);
|
*name = ::tensorflow::io::GetTempFilename(extension);
|
||||||
if (*name.length() == 0) {
|
if (name->length() == 0) {
|
||||||
TF_SetStatus(status, TF_INTERNAL, "Can not create temp file name");
|
TF_SetStatus(status, TF_INTERNAL, "Can not get temp file name");
|
||||||
} else {
|
} else {
|
||||||
TF_SetStatus(status, TF_OK, "");
|
TF_SetStatus(status, TF_OK, "");
|
||||||
}
|
}
|
||||||
|
@ -20,6 +20,8 @@ limitations under the License.
|
|||||||
#include <stddef.h>
|
#include <stddef.h>
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
|
||||||
#include "tensorflow/c/c_api.h"
|
#include "tensorflow/c/c_api.h"
|
||||||
#include "tensorflow/c/tf_file_statistics.h"
|
#include "tensorflow/c/tf_file_statistics.h"
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user