| 873 | } |
| 874 | |
| 875 | inline void ForwardMklTensorInToOut(OpKernelContext* context, int idx_in, |
| 876 | int idx_out) { |
| 877 | int num_inputs = context->num_inputs(); |
| 878 | int num_outputs = context->num_outputs(); |
| 879 | int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); |
| 880 | int idx_meta_in = GetTensorMetaDataIndex(idx_in, num_inputs); |
| 881 | int idx_data_out = GetTensorDataIndex(idx_out, num_outputs); |
| 882 | int idx_meta_out = GetTensorMetaDataIndex(idx_out, num_outputs); |
| 883 | |
| 884 | if (IsRefType(context->input_dtype(idx_data_in))) { |
| 885 | context->forward_ref_input_to_ref_output(idx_data_in, idx_data_out); |
| 886 | context->forward_ref_input_to_ref_output(idx_meta_in, idx_meta_out); |
| 887 | } else { |
| 888 | context->set_output(idx_data_out, context->input(idx_data_in)); |
| 889 | context->set_output(idx_meta_out, context->input(idx_meta_in)); |
| 890 | } |
| 891 | } |
| 892 | |
| 893 | // Set a dummy DNNL shape (called when the output is in TF format) |
| 894 | inline void SetDummyMklDnnShapeOutput(OpKernelContext* context, |
no test coverage detected