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

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h 282 const pooling_backward::primitive_desc& pool_bkwd_prim_desc,
287 pool_bkwd_prim_desc.diff_src_primitive_desc();
mkl_avgpooling_op.cc 624 auto pool_bkwd_prim_desc = pooling_backward::primitive_desc( variable
627 context, pool_bkwd_prim_desc, original_input_dims_nchw,
633 pool_bkwd_prim_desc, &input_gradient_diff_dst, &output_diff_src,
mkl_maxpooling_op.cc 687 auto pool_bkwd_prim_desc = pooling_backward::primitive_desc( variable
690 this->AllocateOutputTensor(context, pool_bkwd_prim_desc,
699 pool_bkwd_prim_desc, &grad_dnn_data, &output_dnn_data,
    [all...]

Completed in 74 milliseconds