Home
last modified time | relevance | path

Searched refs:GetMklShapeList (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_concat_op.cc462 GetMklShapeList(context, "values", &mkl_input_shapes); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h708 inline void GetMklShapeList(OpKernelContext* ctext, StringPiece name, in GetMklShapeList() function