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

  /external/tensorflow/tensorflow/core/kernels/
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 64 milliseconds