We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9e03a7b commit 9ac51a1Copy full SHA for 9ac51a1
tensorflow/core/common_runtime/kernel_benchmark_testlib.cc
@@ -33,21 +33,13 @@ limitations under the License.
33
#include "tensorflow/core/public/version.h"
34
#include "tensorflow/core/util/device_name_utils.h"
35
36
-#if defined(PLATFORM_GOOGLE)
37
-DECLARE_bool(brain_gpu_use_bfc_allocator);
38
-#else
39
-extern bool FLAGS_brain_gpu_use_bfc_allocator;
40
-#endif
41
-
42
namespace tensorflow {
43
namespace test {
44
45
Benchmark::Benchmark(const string& device, Graph* g,
46
const SessionOptions* options, Graph* init) {
47
RequireDefaultOps();
48
49
- FLAGS_brain_gpu_use_bfc_allocator = true;
50
51
SessionOptions default_options;
52
if (!options) {
53
options = &default_options;
0 commit comments