Home
last modified time | relevance | path

Searched refs:GetPrimitiveDesc (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_grad_input_ops.cc123 GetPrimitiveDesc() const { in GetPrimitiveDesc() function in tensorflow::MklConvBwdInputPrimitive
418 auto bwd_input_pd = conv_bwd_input->GetPrimitiveDesc(); in Compute()
Dmkl_conv_grad_filter_ops.cc151 GetPrimitiveDesc() const { in GetPrimitiveDesc() function in tensorflow::MklConvBwdFilterPrimitive
478 auto bwd_filter_pd = conv_bwd_filter->GetPrimitiveDesc(); in Compute()
Dmkl_conv_ops.cc165 std::shared_ptr<ConvFwdPd> GetPrimitiveDesc() const { in GetPrimitiveDesc() function in tensorflow::MklConvFwdPrimitive
1084 std::shared_ptr<ConvFwdPd> conv_fwd_pd = conv_fwd->GetPrimitiveDesc(); in Compute()