Searched refs:input_mkl_tensor (Results 1 – 1 of 1) sorted by relevance
627 const Tensor& input_mkl_tensor, in ConvertMklToTF() argument633 *output_tf_tensor = input_mkl_tensor; in ConvertMklToTF()648 input.SetUsrMem(input_mkl_md, &input_mkl_tensor); in ConvertMklToTF()663 output_tf_tensor->CopyFrom(input_mkl_tensor, output_tf_shape); in ConvertMklToTF()