Searched defs:ConvBwdFilterContext (Results 1 – 1 of 1) sorted by relevance
151 struct ConvBwdFilterContext { struct in tensorflow::MklConvBwdFilterPrimitive153 std::shared_ptr<dnnl::memory> src_mem;154 std::shared_ptr<dnnl::memory> diff_filter_mem;155 std::shared_ptr<dnnl::memory> diff_bias_mem;156 std::shared_ptr<dnnl::memory> diff_dst_mem;159 std::shared_ptr<ConvBwdFilterPd> bwd_filter_pd;160 std::shared_ptr<ConvBwdFilterDesc> bwd_filter_desc;163 std::shared_ptr<ConvFwdPd> fwd_pd;164 std::shared_ptr<ConvFwdDesc> fwd_desc;167 std::shared_ptr<dnnl::primitive> conv_bwd_filter;[all …]