Searched refs:input_mkl_tensor (Results 1 – 1 of 1) sorted by relevance
624 const Tensor& input_mkl_tensor, in ConvertMklToTF() argument630 *output_tf_tensor = input_mkl_tensor; in ConvertMklToTF()645 input.SetUsrMem(input_mkl_md, &input_mkl_tensor); in ConvertMklToTF()660 output_tf_tensor->CopyFrom(input_mkl_tensor, output_tf_shape); in ConvertMklToTF()