Searched defs:SoftmaxFwdContext (Results 1 – 1 of 1) sorted by relevance
94 struct SoftmaxFwdContext { struct in tensorflow::MklSoftmaxPrimitive96 std::shared_ptr<memory> src_mem;97 std::shared_ptr<memory> dst_mem;100 std::shared_ptr<dnnl::softmax_forward::desc> fwd_desc;103 std::shared_ptr<memory::desc> src_md;106 std::shared_ptr<dnnl::softmax_forward::primitive_desc> fwd_pd;107 std::shared_ptr<dnnl::primitive> softmax_fwd;109 std::vector<dnnl::primitive> fwd_primitives;110 std::vector<MemoryArgsMap> fwd_net_args;112 SoftmaxFwdContext() in SoftmaxFwdContext() function