| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_conv_grad_bias_ops.cc | 138 mkl_context.convert_outbackprop, mkl_context.outbackprop_buf,
163 dnnPrimitive_t convert_outbackprop; member in struct:tensorflow::MklConv2DCustomBackpropBiasOp::__anon45424
193 CHECK_EQ(dnnConversionCreate_F32(&convert_outbackprop,
249 if (should_convert_output) dnnDelete_F32(convert_outbackprop);
|
| mkl_maxpooling_op.cc | 310 convert_input = nullptr, convert_outbackprop = nullptr; member in struct:tensorflow::MklMaxPoolingGradOp::__anon45428
380 dnnConversionCreate_F32(&convert_outbackprop, lt_outbackprop_user,
448 if (convert_outbackprop != nullptr) {
451 convert_outbackprop,
456 CHECK_EQ(dnnDelete_F32(convert_outbackprop), E_SUCCESS);
[all...] |
| mkl_avgpooling_op.cc | 278 CHECK_EQ(dnnConversionCreate_F32(&mkl_context.convert_outbackprop,
287 CHECK_EQ(dnnConversionExecute_F32(mkl_context.convert_outbackprop,
292 CHECK_EQ(dnnDelete_F32(mkl_context.convert_outbackprop), E_SUCCESS);
347 dnnPrimitive_t prim_pooling_bwd = nullptr, convert_outbackprop = nullptr; member in struct:tensorflow::MklAvgPoolingGradOp::__anon45423
|
