Searched defs:pooling_res (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dmkl_avgpooling_op.cc116 mkl_context.pooling_res[dnnResourceSrc] = mkl_context.input_buf;
118 mkl_context.pooling_res[dnnResourceSrc] =
139 mkl_context.pooling_res[dnnResourceDst] =
142 mkl_context.pooling_res[dnnResourceWorkspace] = workspace_buf;
145 dnnExecute_F32(mkl_context.prim_pooling_fwd, mkl_context.pooling_res),
159 void* pooling_res[dnnResourceNumber]; member in struct:tensorflow::MklAvgPoolingOp::__anon26513
299 mkl_context.pooling_res[dnnResourceDiffDst] = outbackprop_buf;
301 mkl_context.pooling_res[dnnResourceDiffDst] =
310 mkl_context.pooling_res[dnnResourceWorkspace] = workspace_buf;
332 mkl_context.pooling_res[dnnResourceDiffSr
348 void* pooling_res[dnnResourceNumber]; member in struct:tensorflow::MklAvgPoolingGradOp::__anon26514
[all...]
H A Dmkl_maxpooling_op.cc131 mkl_context.pooling_res[dnnResourceWorkspace] = const_cast<void*>(
133 mkl_context.pooling_res[dnnResourceSrc] =
135 mkl_context.pooling_res[dnnResourceDst] = const_cast<void*>(
139 dnnExecute_F32(mkl_context.prim_pooling_fwd, mkl_context.pooling_res),
149 void* pooling_res[dnnResourceNumber]; member in struct:tensorflow::MklMaxPoolingOp::__anon26526
291 mkl_context.pooling_res[dnnResourceDiffSrc] = const_cast<void*>(
295 dnnExecute_F32(mkl_context.prim_pooling_bwd, mkl_context.pooling_res),
306 void* pooling_res[dnnResourceNumber]; member in struct:tensorflow::MklMaxPoolingGradOp::__anon26527
437 pooling_res[dnnResourceWorkspace] =
441 pooling_res[dnnResourceWorkspac
[all...]

Completed in 69 milliseconds