Home
last modified time | relevance | path

Searched defs:fwd_pd (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_softmax_op.cc98 std::shared_ptr<mkldnn::softmax_forward::primitive_desc> fwd_pd; member
Dmkl_relu_op.cc108 std::shared_ptr<EltwiseFwdPd> fwd_pd; member
146 auto fwd_pd = context_.fwd_pd.get(); in Setup() local
309 std::shared_ptr<EltwiseFwdPd> fwd_pd; member
Dmkl_pooling_ops_common.h123 std::shared_ptr<PoolingFwdPd> fwd_pd; member
269 std::shared_ptr<PoolingFwdPd> fwd_pd; member
Dmkl_conv_grad_input_ops.cc152 std::shared_ptr<ConvFwdPd> fwd_pd; member
Dmkl_fused_batch_norm_op.cc165 std::shared_ptr<BatchNormFwdPd> fwd_pd; member
544 auto fwd_pd = BatchNormFwdPd(fwd_desc, cpu_engine_); in Setup() local
Dmkl_conv_grad_filter_ops.cc173 std::shared_ptr<ConvFwdPd> fwd_pd; member
Dmkl_matmul_ops_common.h148 std::shared_ptr<mkldnn::inner_product_forward::primitive_desc> fwd_pd; member
Dmkl_concat_op.cc328 std::shared_ptr<mkldnn::concat::primitive_desc> fwd_pd; member
Dmkl_conv_ops.cc190 std::shared_ptr<ConvFwdPd> fwd_pd; member