Searched defs:EltwiseFwdContext (Results 1 – 1 of 1) sorted by relevance
107 struct EltwiseFwdContext { struct in tensorflow::MklEltwiseFwdPrimitive109 std::shared_ptr<memory> src_mem;110 std::shared_ptr<memory> dst_mem;113 std::shared_ptr<dnnl::eltwise_forward::desc> fwd_desc;114 std::shared_ptr<EltwiseFwdPd> fwd_pd;117 std::shared_ptr<memory::desc> src_md;118 std::shared_ptr<memory::desc> dst_md;121 std::shared_ptr<dnnl::primitive> eltwise_fwd;123 std::vector<dnnl::primitive> fwd_primitives;125 std::vector<std::unordered_map<int, memory>> fwd_primitives_args;[all …]