Searched defs:src_mem (Results 1 – 3 of 3) sorted by relevance
306 void MKLCpuKernelMod::Reorder(dnnl::memory *src_mem, dnnl::memory *dst_mem) { in Reorder()
287 dnnl::memory src_mem = dnnl::memory(src_desc_, engine_, nullptr); in ComputeMaxValueIndex() local
183 buffer &src_mem, in validate_copy()555 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBuffer() local592 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBufferRect() local802 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBufferToImage() local