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

  /external/tensorflow/tensorflow/core/kernels/
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(&lt_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 &lt_input_prim, prim_pooling_fwd, dnnResourceSrc),
423 &lt_workspace, prim_pooling_fwd, dnnResourceWorkspace)
    [all...]
mkl_avgpooling_op.cc 127 mkl_out_shape.SetMklLayout(mkl_context.prim_pooling_fwd, dnnResourceDst);
145 dnnExecute_F32(mkl_context.prim_pooling_fwd, mkl_context.pooling_res),
155 dnnPrimitive_t prim_pooling_fwd = nullptr, convert_input = nullptr; member in struct:tensorflow::MklAvgPoolingOp::__anon39834
178 &prim_pooling_fwd, primAttr, algorithm, lt_user_input,
184 &lt_prim_input, prim_pooling_fwd, dnnResourceSrc),
195 CHECK_EQ(dnnLayoutCreateFromPrimitive_F32(&lt_workspace, prim_pooling_fwd,
206 CHECK_EQ(dnnDelete_F32(prim_pooling_fwd), E_SUCCESS);

Completed in 62 milliseconds