Searched refs:reorder_pd (Results 1 – 4 of 4) sorted by relevance
150 auto reorder_pd = in Compute() local153 net.push_back(reorder(reorder_pd)); in Compute()
111 std::shared_ptr<reorder::primitive_desc> reorder_pd; member122 reorder_pd(nullptr), in ReorderContext()144 context_.reorder_pd.reset( in Setup()149 context_.reorder_prim.reset(new reorder(*context_.reorder_pd)); in Setup()
128 mkldnn::reorder::primitive_desc reorder_pd = in Compute() local137 new mkldnn::reorder(reorder_pd)); in Compute()
217 std::shared_ptr<reorder::primitive_desc> reorder_pd; member237 context_.reorder_pd = std::make_shared<reorder::primitive_desc>( in Setup()240 std::make_shared<mkldnn::reorder>(reorder(*context_.reorder_pd)); in Setup()