Searched defs:MklCreateLayoutsAndPrimitives (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ | ||
H A D | mkl_avgpooling_op.cc | 92 mkl_context.MklCreateLayoutsAndPrimitives(context, 161 void MklCreateLayoutsAndPrimitives(OpKernelContext* context, function in struct:tensorflow::MklAvgPoolingOp::__anon26513 272 mkl_context.MklCreateLayoutsAndPrimitives(context); 352 void MklCreateLayoutsAndPrimitives(OpKernelContext* context) { function in struct:tensorflow::MklAvgPoolingGradOp::__anon26514 |
H A D | mkl_maxpooling_op.cc | 100 mkl_context.MklCreateLayoutsAndPrimitives(context); 153 void MklCreateLayoutsAndPrimitives(OpKernelContext* context) { function in struct:tensorflow::MklMaxPoolingOp::__anon26526 |
Completed in 121 milliseconds