Searched refs:output_md (Results 1 – 2 of 2) sorted by relevance
105 memory::desc output_md = in Compute() local126 new memory(output_md, cpu_engine_, output_buf)); in Compute()
440 auto output_md = in ComputeMklSlice() local442 auto output_pd = output_md; in ComputeMklSlice()447 output.SetUsrMem(output_md, output_tensor); in ComputeMklSlice()