diff options
Diffstat (limited to 'tensorflow/core/ops/array_ops.cc')
| -rw-r--r-- | tensorflow/core/ops/array_ops.cc | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/tensorflow/core/ops/array_ops.cc b/tensorflow/core/ops/array_ops.cc index cf68680bb3..a1d09bd503 100644 --- a/tensorflow/core/ops/array_ops.cc +++ b/tensorflow/core/ops/array_ops.cc @@ -1542,6 +1542,23 @@ REGISTER_OP("Identity") Return a tensor with the same shape and contents as the input tensor or value. )Doc"); +#ifdef INTEL_MKL +REGISTER_OP("_MklIdentity") + .Input("input: T") + .Input("mkl_input: uint8") + .Output("output: T") + .Output("mkl_output: uint8") + .Attr("T: type") + .SetShapeFn([](shape_inference::InferenceContext* c) { + c->set_output(0, c->input(0)); + c->set_output_handle_dtype(0, c->input_handle_dtype(0)); + c->set_output_handle_shape(0, c->input_handle_shape(0)); + return Status::OK(); + }) + .Doc(R"Doc( Mkl implementation of IdentityOp +)Doc"); +#endif + // -------------------------------------------------------------------------- REGISTER_OP("RefIdentity") .Input("input: Ref(T)") |