Home
last modified time | relevance | path

Searched refs:conv_res (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_grad_bias_ops.cc134 dnnExecute_F32(mkl_context.prim_conv_bwdbias, mkl_context.conv_res), in Compute()
160 void* conv_res[dnnResourceNumber]; member
199 conv_res[dnnResourceDiffBias] = outbackprop_buf; in MklPrepareConvolutionOutputs()
201 conv_res[dnnResourceDiffBias] = in MklPrepareConvolutionOutputs()
233 conv_res[dnnResourceDiffDst] = input_buf; in MklPrepareConvolutionInputs()
236 conv_res[dnnResourceDiffDst] = in MklPrepareConvolutionInputs()
Dmkl_conv_ops.cc665 mkl_context.conv_res[dnnResourceDst] = in Compute()
678 CHECK_EQ(dnnExecute_F32(mkl_context.prim_fwd, mkl_context.conv_res), in Compute()
700 void* conv_res[dnnResourceNumber]; member
769 conv_res[dnnResourceSrc] = in MklPrepareConvolutionInputs()
796 conv_res[dnnResourceFilter] = in MklPrepareConvolutionInputs()
820 conv_res[dnnResourceBias] = in MklPrepareConvolutionInputs()