| 40 | class DebugIdentityOpTest : public OpsTestBase { |
| 41 | protected: |
| 42 | Status Init(DataType input_type, const std::vector<string>& debug_urls) { |
| 43 | env_ = Env::Default(); |
| 44 | |
| 45 | TF_CHECK_OK(NodeDefBuilder("op", "DebugIdentity") |
| 46 | .Input(FakeInput(input_type)) |
| 47 | .Attr("tensor_name", "FakeTensor:0") |
| 48 | .Attr("debug_urls", debug_urls) |
| 49 | .Finalize(node_def())); |
| 50 | return InitOp(); |
| 51 | } |
| 52 | |
| 53 | Status Init(DataType input_type) { |
| 54 | std::vector<string> empty_debug_urls; |