Searched refs:primAttr (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | mkl_maxpooling_op.cc | 165 dnnPrimitiveAttributes_t primAttr = nullptr; local
169 &prim_pooling_fwd, primAttr, algorithm, lt_user_input,
343 dnnPrimitiveAttributes_t primAttr = nullptr; local
347 &prim_pooling_fwd, primAttr, algorithm, lt_input_user,
354 &prim_pooling_bwd, primAttr, algorithm, lt_input_user,
|
| H A D | mkl_avgpooling_op.cc | 174 dnnPrimitiveAttributes_t primAttr = nullptr; local
178 &prim_pooling_fwd, primAttr, algorithm, lt_user_input,
396 dnnPrimitiveAttributes_t primAttr = nullptr; local
398 &prim_pooling_bwd, primAttr, algorithm, lt_user_input,
|
Completed in 164 milliseconds