Searched refs:pool_desc (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dmkl_avgpooling_op.cc507 auto pool_desc = pooling_forward::desc( variable
518 pooling_forward::primitive_desc(pool_desc, cpu_engine);
H A Dmkl_maxpooling_op.cc551 auto pool_desc = pooling_forward::desc( variable
562 pooling_forward::primitive_desc(pool_desc, cpu_engine);

Completed in 167 milliseconds