HomeSort by relevance Sort by last modified time
    Searched refs:input_mkl_md (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_tfconv_op.h 93 auto input_mkl_md = input_shape.GetMklLayout(); local
99 input.SetUsrMem(input_mkl_md, &input_tensor);
mkl_reshape_op.cc 153 auto input_mkl_md = mkl_shape_input.GetMklLayout(); local
154 if (mkl_shape_input.GetTfDataFormat() == input_mkl_md.data.format) {
244 auto input_mkl_md = mkl_shape_input.GetMklLayout(); variable
246 dnn_data_input.SetUsrMem(input_mkl_md, &input_tensor);

Completed in 340 milliseconds