Searched refs:GetMklInputList (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_concat_op.cc | 159 GetMklInputList(context, "values", &input_tensors); in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 904 inline void GetMklInputList(OpKernelContext* ctext, StringPiece name, in GetMklInputList() function
915 GetMklInputList(ctext, strings::StrCat("mkl_", name), &input_mkl_tensors); in GetMklShapeList()
929 GetMklInputList(ctext, strings::StrCat("mkl_", name), &input_mkl_tensors); in GetMklShapeList()
|