Searched defs:ConvFwdContext (Results 1 – 1 of 1) sorted by relevance
217 struct ConvFwdContext { struct in tensorflow::MklConvFwdPrimitive219 std::shared_ptr<dnnl::memory> src_mem;220 std::shared_ptr<dnnl::memory> filter_mem;221 std::shared_ptr<dnnl::memory> bias_mem;222 std::shared_ptr<dnnl::memory> dst_mem;223 std::shared_ptr<dnnl::memory> sp_mem;226 std::shared_ptr<dnnl::memory> bn_scale_mem;227 std::shared_ptr<dnnl::memory> bn_mean_mem;228 std::shared_ptr<dnnl::memory> bn_rsqrt_mem;229 std::shared_ptr<dnnl::memory> bn_offset_mem;[all …]