| 856 | } |
| 857 | |
| 858 | inline void ForwardTfTensorInToOut(OpKernelContext* context, int idx_in, |
| 859 | int idx_out) { |
| 860 | int num_inputs = context->num_inputs(); |
| 861 | int num_outputs = context->num_outputs(); |
| 862 | int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); |
| 863 | int idx_data_out = GetTensorDataIndex(idx_out, num_outputs); |
| 864 | |
| 865 | MklDnnShape dnn_shape_output; |
| 866 | dnn_shape_output.SetMklTensor(false); |
| 867 | AllocateOutputSetMklShape(context, idx_out, dnn_shape_output); |
| 868 | if (IsRefType(context->input_dtype(idx_data_in))) { |
| 869 | context->forward_ref_input_to_ref_output(idx_data_in, idx_data_out); |
| 870 | } else { |
| 871 | context->set_output(idx_data_out, context->input(idx_data_in)); |
| 872 | } |
| 873 | } |
| 874 | |
| 875 | inline void ForwardMklTensorInToOut(OpKernelContext* context, int idx_in, |
| 876 | int idx_out) { |
no test coverage detected