Searched refs:pool_fwd_prim_desc (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | mkl_maxpooling_op.cc | 588 const pooling_forward::primitive_desc& pool_fwd_prim_desc, 593 pool_fwd_prim_desc.workspace_primitive_desc(); 674 auto pool_fwd_prim_desc = variable 688 pool_bkwd_desc, cpu_engine, pool_fwd_prim_desc); 696 pool_fwd_prim_desc.workspace_primitive_desc(), 586 AllocateWorkspaceTensor( OpKernelContext* context, const pooling_forward::primitive_desc& pool_fwd_prim_desc, MklDnnData<uint8>* dnn_data_wksp) argument
|
H A D | mkl_pooling_ops_common.h | 215 const pooling_forward::primitive_desc& pool_fwd_prim_desc, 219 memory::primitive_desc dst_pd = pool_fwd_prim_desc.dst_primitive_desc(); 213 AllocateOutputTensor( OpKernelContext* context, const pooling_forward::primitive_desc& pool_fwd_prim_desc, const memory::dims output_dims_mkl_order, const memory::format& output_tf_format, Tensor** output_tensor) argument
|
H A D | mkl_avgpooling_op.cc | 611 auto pool_fwd_prim_desc = variable 625 pool_bkwd_desc, cpu_engine, pool_fwd_prim_desc);
|
Completed in 90 milliseconds