| 817 | } |
| 818 | |
| 819 | inline void CopyMklTensorInToOut(OpKernelContext* context, int idx_in, |
| 820 | int idx_out) { |
| 821 | int num_inputs = context->num_inputs(); |
| 822 | int num_outputs = context->num_outputs(); |
| 823 | int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); |
| 824 | int idx_meta_in = GetTensorMetaDataIndex(idx_in, num_inputs); |
| 825 | int idx_data_out = GetTensorDataIndex(idx_out, num_outputs); |
| 826 | int idx_meta_out = GetTensorMetaDataIndex(idx_out, num_outputs); |
| 827 | |
| 828 | const Tensor& data = context->input(idx_data_in); |
| 829 | const Tensor& meta = context->input(idx_meta_in); |
| 830 | Tensor output(data.dtype()); |
| 831 | Tensor meta_output(meta.dtype()); |
| 832 | |
| 833 | // TODO(intel_tf): alternatively, call forward_input_to_output_with_shape(...) |
| 834 | CHECK(output.CopyFrom(data, data.shape())); |
| 835 | CHECK(meta_output.CopyFrom(meta, meta.shape())); |
| 836 | context->set_output(idx_data_out, output); |
| 837 | context->set_output(idx_meta_out, meta_output); |
| 838 | } |
| 839 | |
| 840 | inline void CopyTfTensorInToOutWithShape(OpKernelContext* context, int idx_in, |
| 841 | int idx_out, |
no test coverage detected