Home
last modified time | relevance | path

Searched defs:src_mem (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc70 void *src_mem = const_cast<void *>(gpu_src.opaque()); in Memcpy() local
91 void *src_mem = const_cast<void *>(gpu_src.opaque()); in MemcpyDeviceToDevice() local
/external/mesa3d/src/gallium/state_trackers/clover/api/
Dtransfer.cpp137 buffer &src_mem, in validate_copy()
427 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBuffer() local
464 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBufferRect() local
642 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBufferToImage() local
/external/tensorflow/tensorflow/core/kernels/
Dmkl_slice_op.cc206 std::shared_ptr<mkldnn::memory> src_mem; member
Dmkl_fused_batch_norm_op.cc117 std::shared_ptr<mkldnn::memory> src_mem; member
359 std::shared_ptr<mkldnn::memory> src_mem; member
Dmkl_relu_op.cc103 std::shared_ptr<memory> src_mem; member
294 std::shared_ptr<memory> src_mem; member
Dmkl_conv_grad_filter_ops.cc169 std::shared_ptr<mkldnn::memory> src_mem; member
Dmkl_pooling_ops_common.h117 std::shared_ptr<mkldnn::memory> src_mem; member
Dmkl_conv_ops.cc177 std::shared_ptr<mkldnn::memory> src_mem; member
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h2274 std::shared_ptr<mkldnn::memory> src_mem; member