Searched refs:pool_fwd_prim_desc (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dmkl_maxpooling_op.cc588 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 Dmkl_pooling_ops_common.h215 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 Dmkl_avgpooling_op.cc611 auto pool_fwd_prim_desc = variable
625 pool_bkwd_desc, cpu_engine, pool_fwd_prim_desc);

Completed in 90 milliseconds