mkl_maxpooling_op.cc | 278 mkl_output_shape.SetMklLayout(mkl_context.prim_pooling_bwd, 295 dnnExecute_F32(mkl_context.prim_pooling_bwd, mkl_context.pooling_res), 307 dnnPrimitive_t prim_pooling_fwd = nullptr, prim_pooling_bwd = nullptr, member in struct:tensorflow::MklMaxPoolingGradOp::__anon39848 354 &prim_pooling_bwd, primAttr, algorithm, lt_input_user, 361 <_outbackprop_prim, prim_pooling_bwd, dnnResourceDiffDst), 475 CHECK_EQ(dnnDelete_F32(prim_pooling_bwd), E_SUCCESS); [all...] |