Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:pooling_res (Results 1 - 2 of 2) sorted by null

/external/tensorflow/tensorflow/core/kernels/
mkl_avgpooling_op.cc	116 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::__anon39834 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[dnnResourceDiffSrc] 348 void* pooling_res[dnnResourceNumber]; member in struct:tensorflow::MklAvgPoolingGradOp::__anon39835     [all...]
mkl_maxpooling_op.cc	131 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::__anon39847 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::__anon39848 437 pooling_res[dnnResourceWorkspace] = 441 pooling_res[dnnResourceWorkspace] = const_cast<void*>     [all...]

Completed in 82 milliseconds