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::__anon39835
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 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 &lt_outbackprop_prim, prim_pooling_bwd, dnnResourceDiffDst),
475 CHECK_EQ(dnnDelete_F32(prim_pooling_bwd), E_SUCCESS);
    [all...]

Completed in 60 milliseconds