experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixing Windows build.

Browse Source 
PiperOrigin-RevId: 342070912
Change-Id: I293a6865466ca5cbddc2d7d18e451b78a36fd9d7
This commit is contained in:
Roman Dzhabarov 2020-11-12 09:59:21 -08:00 committed by TensorFlower Gardener
parent 3ea78d0fdd
commit 76b18768c3
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc
View File

@ -28,14 +28,13 @@ using ::testing::Test;
class MockMlirOptimizationPass : public MlirOptimizationPass {
public:
MOCK_METHOD(llvm::StringRef, name, (), (const, override));
MOCK_METHOD(bool, IsEnabled,
(const ConfigProto& config_proto, const Graph& graph),
(const, override));
MOCK_METHOD(Status, Run,
(const ConfigProto& config_proto, mlir::ModuleOp module,
const Graph& graph),
(override));
// MOCK_METHOD does not work on Windows build, using MOCK_METHODX
// instead.
MOCK_CONST_METHOD0(name, llvm::StringRef());
MOCK_CONST_METHOD2(IsEnabled,
bool(const ConfigProto& config_proto, const Graph& graph));
MOCK_METHOD3(Run, Status(const ConfigProto& config_proto,
mlir::ModuleOp module, const Graph& graph));
};
class MlirGraphOptimizationPassTest : public Test {