HomeSort by relevance Sort by last modified time
    Searched refs:prim_pooling_bwd (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_avgpooling_op.cc 317 mkl_out_shape.SetMklLayout(mkl_context.prim_pooling_bwd,
337 dnnExecute_F32(mkl_context.prim_pooling_bwd, mkl_context.pooling_res),
347 dnnPrimitive_t prim_pooling_bwd = nullptr, convert_outbackprop = nullptr; member in struct:tensorflow::MklAvgPoolingGradOp::__anon45423
398 &prim_pooling_bwd, primAttr, algorithm, lt_user_input,
405 &lt_prim_outbackprop, prim_pooling_bwd, dnnResourceDiffDst),
408 CHECK_EQ(dnnLayoutCreateFromPrimitive_F32(&lt_workspace, prim_pooling_bwd,
415 CHECK_EQ(dnnDelete_F32(prim_pooling_bwd), E_SUCCESS);
mkl_maxpooling_op.cc 280 mkl_output_shape.SetMklLayout(mkl_context.prim_pooling_bwd,
297 dnnExecute_F32(mkl_context.prim_pooling_bwd, mkl_context.pooling_res),
309 dnnPrimitive_t prim_pooling_fwd = nullptr, prim_pooling_bwd = nullptr, member in struct:tensorflow::MklMaxPoolingGradOp::__anon45428
356 &prim_pooling_bwd, primAttr, algorithm, lt_input_user,
363 &lt_outbackprop_prim, prim_pooling_bwd, dnnResourceDiffDst),
477 CHECK_EQ(dnnDelete_F32(prim_pooling_bwd), E_SUCCESS);
    [all...]

Completed in 74 milliseconds