From f41959ccb2d9d4c722fe8fc3351401d53bcf4900 Mon Sep 17 00:00:00 2001 From: Manjunath Kudlur Date: Fri, 6 Nov 2015 16:27:58 -0800 Subject: TensorFlow: Initial commit of TensorFlow library. TensorFlow is an open source software library for numerical computation using data flow graphs. Base CL: 107276108 --- tensorflow/core/ops/logging_ops.cc | 43 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 43 insertions(+) create mode 100644 tensorflow/core/ops/logging_ops.cc (limited to 'tensorflow/core/ops/logging_ops.cc') diff --git a/tensorflow/core/ops/logging_ops.cc b/tensorflow/core/ops/logging_ops.cc new file mode 100644 index 0000000000..28546fe645 --- /dev/null +++ b/tensorflow/core/ops/logging_ops.cc @@ -0,0 +1,43 @@ +#include "tensorflow/core/framework/op.h" + +namespace tensorflow { + +REGISTER_OP("Assert") + .Input("condition: bool") + .Input("data: T") + .Attr("T: list(type)") + .Attr("summarize: int = 3") + .Doc(R"doc( +Asserts that the given condition is true. + +If `condition` evaluates to false, print the list of tensors in `data`. +`summarize` determines how many entries of the tensors to print. + +condition: The condition to evaluate. +data: The tensors to print out when condition is false. +summarize: Print this many entries of each tensor. +)doc"); + +REGISTER_OP("Print") + .Input("input: T") + .Input("data: U") + .Output("output: T") + .Attr("T: type") + .Attr("U: list(type)") + .Attr("message: string = ''") + .Attr("first_n: int = -1") + .Attr("summarize: int = 3") + .Doc(R"doc( +Prints a list of tensors. + +Passes `input` through to `output` and prints `data` when evaluating. + +input: The tensor passed to `output` +data: A list of tensors to print out when op is evaluated. +output:= The unmodified `input` tensor +message: A string, prefix of the error message. +first_n: Only log `first_n` number of times. -1 disables logging. +summarize: Only print this many entries of each tensor. +)doc"); + +} // end namespace tensorflow -- cgit v1.2.3