Searched refs:idx_data_in (Results 1 – 1 of 1) sorted by relevance
817 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in CopyMklTensorInToOut() local822 const Tensor& data = context->input(idx_data_in); in CopyMklTensorInToOut()839 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in CopyTfTensorInToOutWithShape() local842 const Tensor& data = context->input(idx_data_in); in CopyTfTensorInToOutWithShape()856 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in ForwardTfTensorInToOut() local862 if (IsRefType(context->input_dtype(idx_data_in))) { in ForwardTfTensorInToOut()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()873 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in ForwardMklTensorInToOut() local878 if (IsRefType(context->input_dtype(idx_data_in))) { in ForwardMklTensorInToOut()[all …]