diff options
Diffstat (limited to 'res/TensorFlowTests/NET_0015/test.pbtxt')
-rw-r--r-- | res/TensorFlowTests/NET_0015/test.pbtxt | 74 |
1 files changed, 74 insertions, 0 deletions
diff --git a/res/TensorFlowTests/NET_0015/test.pbtxt b/res/TensorFlowTests/NET_0015/test.pbtxt new file mode 100644 index 000000000..3d994e8d5 --- /dev/null +++ b/res/TensorFlowTests/NET_0015/test.pbtxt @@ -0,0 +1,74 @@ +# A simple network that simulates Rsqrt: 1.0 / Sqrt(input) +node { + name: "Placeholder" + op: "Placeholder" + attr { + key: "dtype" + value { + type: DT_FLOAT + } + } + attr { + key: "shape" + value { + shape { + dim { + size: 2 + } + dim { + size: 3 + } + } + } + } +} +node { + name: "SQRT_02" + op: "Sqrt" + input: "Placeholder" + attr { + key: "T" + value { + type: DT_FLOAT + } + } +} +node { + name: "Const" + op: "Const" + attr { + key: "dtype" + value { + type: DT_FLOAT + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_FLOAT + tensor_shape { + dim { + size: 2 + } + dim { + size: 3 + } + } + float_val: 1.0 + } + } + } +} +node { + name: "as_rsqrt" + op: "RealDiv" + input: "Const" + input: "SQRT_02" + attr { + key: "T" + value { + type: DT_FLOAT + } + } +} |