Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_input_ops.cc135 std::shared_ptr<ConvBwdDataPd> GetPrimitiveDesc() const { in GetPrimitiveDesc() function in tensorflow::MklConvBwdInputPrimitive
437 auto bwd_input_pd = conv_bwd_input->GetPrimitiveDesc(); in Compute()
Dmkl_matmul_op_fused.cc132 matmul_prim->GetPrimitiveDesc(); in Compute()
Dmkl_conv_grad_filter_ops.cc155 std::shared_ptr<ConvBwdFilterPd> GetPrimitiveDesc() const { in GetPrimitiveDesc() function in tensorflow::MklConvBwdFilterPrimitive
540 auto bwd_filter_pd = conv_bwd_filter->GetPrimitiveDesc(); in Compute()
Dmkl_qmatmul_op.cc240 matmul_fwd_pd = matmul_fwd->GetPrimitiveDesc(); in Compute()
Dmkl_matmul_ops_common.h133 GetPrimitiveDesc() const { in GetPrimitiveDesc() function
Dmkl_conv_ops.cc167 std::shared_ptr<ConvFwdPd> GetPrimitiveDesc() const { in GetPrimitiveDesc() function in tensorflow::MklConvFwdPrimitive
665 std::shared_ptr<ConvFwdPd> conv_fwd_pd = conv_fwd->GetPrimitiveDesc(); in Compute()