Searched refs:prim_fwd (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.cc | 610 &mkl_context.prim_fwd, nullptr, dnnAlgorithmConvolutionDirect, in Compute()
618 &mkl_context.prim_fwd, nullptr, dnnAlgorithmConvolutionDirect, in Compute()
628 mkl_output_mkl_shape.SetMklLayout(mkl_context.prim_fwd, dnnResourceDst); in Compute()
645 mkl_filter_output_mkl_shape.SetMklLayout(mkl_context.prim_fwd, in Compute()
678 CHECK_EQ(dnnExecute_F32(mkl_context.prim_fwd, mkl_context.conv_res), in Compute()
699 dnnPrimitive_t prim_fwd; member
752 prim_fwd, dnnResourceSrc), in MklPrepareConvolutionInputs()
776 prim_fwd, dnnResourceFilter), in MklPrepareConvolutionInputs()
804 prim_fwd, dnnResourceBias), in MklPrepareConvolutionInputs()
827 dnnDelete_F32(prim_fwd); in MklCleanup()
|