Searched refs:idx_data_out (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 819 int idx_data_out = GetTensorDataIndex(idx_out, num_outputs); in CopyMklTensorInToOut() local
830 context->set_output(idx_data_out, output); in CopyMklTensorInToOut()
840 int idx_data_out = GetTensorDataIndex(idx_out, num_outputs); in CopyTfTensorInToOutWithShape() local
849 context->set_output(idx_data_out, output); in CopyTfTensorInToOutWithShape()
857 int idx_data_out = GetTensorDataIndex(idx_out, num_outputs); in ForwardTfTensorInToOut() local
863 context->forward_ref_input_to_ref_output(idx_data_in, idx_data_out); in ForwardTfTensorInToOut()
865 context->set_output(idx_data_out, context->input(idx_data_in)); in ForwardTfTensorInToOut()
875 int idx_data_out = GetTensorDataIndex(idx_out, num_outputs); in ForwardMklTensorInToOut() local
879 context->forward_ref_input_to_ref_output(idx_data_in, idx_data_out); in ForwardMklTensorInToOut()
882 context->set_output(idx_data_out, context->input(idx_data_in)); in ForwardMklTensorInToOut()
[all …]
|