diff options
| author |  Gunhan Gulsoy <gunan@google.com> | 2018-08-14 23:38:40 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-14 23:42:02 -0700 |
| commit | 5d9d5b41615259c1173bc9f6a086668fd7185bad (patch) | |
| tree | e5f6268293bf2d1c61e7ba4605c5f23a55e0e9cc /tensorflow/core/graph | |
| parent | 63af3e170e0022c3e40d53dbc25c9d6d54f7c7be (diff) | |
Automated rollback of commit a00fba38681fa28d6d20cd730fe362864b819a0d
PiperOrigin-RevId: 208775205
Diffstat (limited to 'tensorflow/core/graph')
| -rw-r--r-- | tensorflow/core/graph/testlib.cc | 25 |
1 files changed, 0 insertions, 25 deletions
diff --git a/tensorflow/core/graph/testlib.cc b/tensorflow/core/graph/testlib.cc index 67b252cb6c..ea7788f654 100644 --- a/tensorflow/core/graph/testlib.cc +++ b/tensorflow/core/graph/testlib.cc @@ -21,39 +21,14 @@ limitations under the License. #include "tensorflow/core/framework/node_def_builder.h" #include "tensorflow/core/framework/node_def_util.h" #include "tensorflow/core/framework/op.h" -#include "tensorflow/core/framework/op_kernel.h" #include "tensorflow/core/framework/types.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/graph/graph.h" #include "tensorflow/core/graph/node_builder.h" -#include "tensorflow/core/kernels/constant_op.h" #include "tensorflow/core/lib/core/status.h" #include "tensorflow/core/platform/logging.h" namespace tensorflow { - -// HostConst: forced to generate output on the host. -// Only used by testlib; no op is registered for this kernel -// externally (i.e., in array_ops.cc) -REGISTER_KERNEL_BUILDER(Name("HostConst").Device(DEVICE_CPU), HostConstantOp); -REGISTER_KERNEL_BUILDER( - Name("HostConst").Device(DEVICE_GPU).HostMemory("output"), HostConstantOp); -#ifdef TENSORFLOW_USE_SYCL -REGISTER_KERNEL_BUILDER( - Name("HostConst").Device(DEVICE_SYCL).HostMemory("output"), HostConstantOp); -#endif // TENSORFLOW_USE_SYCL - -// Register the HostConst Op -// Returns a constant tensor on the host. Useful for writing C++ tests -// and benchmarks which run on GPU but require arguments pinned to the host. -// Used by test::graph::HostConstant. -// value: Attr `value` is the tensor to return. -REGISTER_OP("HostConst") - .Output("output: dtype") - .Attr("value: tensor") - .Attr("dtype: type") - .SetShapeFn(shape_inference::UnknownShape); - namespace test { namespace graph { |