MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / MakeOp

Method MakeOp

tensorflow/core/kernels/save_op_test.cc:45–55  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

43class SaveOpTest : public OpsTestBase {
44 protected:
45 void MakeOp() {
46 TF_ASSERT_OK(
47 NodeDefBuilder("myop", "Save")
48 .Input(FakeInput())
49 .Input(FakeInput())
50 .Input(FakeInput({DT_BOOL, DT_INT32, DT_FLOAT, DT_DOUBLE, DT_QINT8,
51 DT_QINT32, DT_UINT8, DT_INT8, DT_INT16, DT_INT64,
52 DT_STRING, DT_COMPLEX64, DT_COMPLEX128, DT_HALF}))
53 .Finalize(node_def()));
54 TF_ASSERT_OK(InitOp());
55 }
56};
57
58TEST_F(SaveOpTest, Simple) {

Callers

nothing calls this directly

Calls 4

NodeDefBuilderClass · 0.85
FakeInputFunction · 0.50
FinalizeMethod · 0.45
InputMethod · 0.45

Tested by

no test coverage detected