Searched refs:mkl_tensor (Results 1 – 2 of 2) sorted by relevance
189 const Tensor* mkl_tensor; in Compute() local195 mkl_tensor = &input_tensor_0; in Compute()201 mkl_tensor = &input_tensor_1; in Compute()234 mkl_tensor->shape(), mkl_output_mkl_shape); in Compute()
803 inline Tensor ConvertMklToTF(OpKernelContext* context, const Tensor& mkl_tensor, in ConvertMklToTF() argument817 void* input_buffer = const_cast<T*>(mkl_tensor.flat<T>().data()); in ConvertMklToTF()820 if (mkl_tensor.NumElements() != 0) { in ConvertMklToTF()832 inline Tensor ConvertMklToTF(OpKernelContext* context, const Tensor& mkl_tensor, in ConvertMklToTF() argument837 return mkl_tensor; // return input since it is already TF tensor in ConvertMklToTF()852 input.SetUsrMem(input_mkl_md, &mkl_tensor); in ConvertMklToTF()862 CHECK(output_tensor.CopyFrom(mkl_tensor, output_shape)); in ConvertMklToTF()