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

  /external/tensorflow/tensorflow/core/kernels/
mkl_avgpooling_op.cc 611 auto pool_fwd_prim_desc = variable
625 pool_bkwd_desc, cpu_engine, pool_fwd_prim_desc);
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(),
    [all...]

Completed in 155 milliseconds