Searched defs:mkl_shape (Results 1 – 2 of 2) sorted by relevance
190 const MklDnnShape* mkl_shape; in Compute() local
804 const MklShape& mkl_shape) { in ConvertMklToTF()833 const MklDnnShape& mkl_shape) { in ConvertMklToTF()964 const MklShape& mkl_shape) { in AllocateOutputSetMklShape()980 const MklDnnShape& mkl_shape) { in AllocateOutputSetMklShape()999 const MklShape& mkl_shape) { in AllocateOutputSetMklShape()1020 const MklDnnShape& mkl_shape) { in AllocateOutputSetMklShape()1090 const MklShape& mkl_shape, in MklSizesToTFSizes()1132 inline int64 GetMklTensorDim(const MklShape& mkl_shape, char dimension) { in GetMklTensorDim()1268 const MklDnnShape& mkl_shape) { in ForwardMklTensorInToOutWithMklShape()