blob: 39b38e9a59bfcc5a3b4b4bf65bd13dd86c4ebd07 (
plain)
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
|
#ifndef TENSORFLOW_FRAMEWORK_FAKE_INPUT_H_
#define TENSORFLOW_FRAMEWORK_FAKE_INPUT_H_
#include "tensorflow/core/framework/node_def_builder.h"
#include "tensorflow/core/framework/types.h"
namespace tensorflow {
// These functions return values that may be passed to
// NodeDefBuilder::Input() to add an input for a test. Use them when
// you don't care about the node names/output indices providing the
// input. They also allow you to omit the input types and/or
// list length when they may be inferred.
FakeInputFunctor FakeInput(); // Infer everything
FakeInputFunctor FakeInput(DataType dt);
FakeInputFunctor FakeInput(int n); // List of length n
FakeInputFunctor FakeInput(int n, DataType dt);
FakeInputFunctor FakeInput(DataTypeSlice dts);
inline FakeInputFunctor FakeInput(std::initializer_list<DataType> dts) {
return FakeInput(DataTypeSlice(dts));
}
} // namespace tensorflow
#endif // TENSORFLOW_FRAMEWORK_FAKE_INPUT_H_
|