mkl_maxpooling_op.cc | 107 mkl_out_shape.SetMklLayout(mkl_context.prim_pooling_fwd, dnnResourceDst); 139 dnnExecute_F32(mkl_context.prim_pooling_fwd, mkl_context.pooling_res), 150 dnnPrimitive_t prim_pooling_fwd = nullptr; member in struct:tensorflow::MklMaxPoolingOp::__anon39847 169 &prim_pooling_fwd, primAttr, algorithm, lt_user_input, 175 CHECK_EQ(dnnLayoutCreateFromPrimitive_F32(<_workspace, prim_pooling_fwd, 182 CHECK_EQ(dnnDelete_F32(prim_pooling_fwd), E_SUCCESS); 307 dnnPrimitive_t prim_pooling_fwd = nullptr, prim_pooling_bwd = nullptr, member in struct:tensorflow::MklMaxPoolingGradOp::__anon39848 347 &prim_pooling_fwd, primAttr, algorithm, lt_input_user, 366 <_input_prim, prim_pooling_fwd, dnnResourceSrc), 423 <_workspace, prim_pooling_fwd, dnnResourceWorkspace) [all...] |