Home
last modified time | relevance | path

Searched refs:input_mkl_md (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_reshape_op.cc54 auto input_mkl_md = mkl_shape_input.GetMklLayout(); in SkipReorder() local
55 if (mkl_shape_input.GetTfDataFormat() == input_mkl_md.data.format && in SkipReorder()
146 auto input_mkl_md = mkl_shape_input.GetMklLayout(); in Compute() local
148 dnn_data_input.SetUsrMem(input_mkl_md, &input_tensor); in Compute()
Dmkl_tfconv_op.h89 auto input_mkl_md = input_shape.GetMklLayout(); in ConvertMklToTf() local
95 input.SetUsrMem(input_mkl_md, &input_tensor); in ConvertMklToTf()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h849 auto input_mkl_md = mkl_shape.GetMklLayout(); in ConvertMklToTF() local
852 input.SetUsrMem(input_mkl_md, &mkl_tensor); in ConvertMklToTF()