Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc103 void *src_mem = const_cast<void *>(gpu_src.opaque()); in Memcpy() local
124 void *src_mem = const_cast<void *>(gpu_src.opaque()); in MemcpyDeviceToDevice() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_softmax_op.cc88 std::shared_ptr<memory> src_mem; member
Dmkl_quantize_op.cc107 std::shared_ptr<mkldnn::memory> src_mem; member
Dmkl_slice_op.cc214 std::shared_ptr<mkldnn::memory> src_mem; member
Dmkl_relu_op.cc103 std::shared_ptr<memory> src_mem; member
289 std::shared_ptr<memory> src_mem; member
Dmkl_fused_batch_norm_op.cc160 std::shared_ptr<mkldnn::memory> src_mem; member
498 std::shared_ptr<mkldnn::memory> src_mem; member
Dmkl_conv_grad_filter_ops.cc163 std::shared_ptr<mkldnn::memory> src_mem; member
Dmkl_matmul_ops_common.h146 std::shared_ptr<mkldnn::memory> src_mem; member
Dmkl_concat_op.cc348 std::shared_ptr<mkldnn::memory> src_mem( in Setup() local
Dmkl_pooling_ops_common.h118 std::shared_ptr<mkldnn::memory> src_mem; member
Dmkl_conv_ops.cc217 std::shared_ptr<mkldnn::memory> src_mem; member
/external/mesa3d/src/gallium/frontends/clover/api/
Dtransfer.cpp159 buffer &src_mem, in validate_copy()
518 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBuffer() local
555 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBufferRect() local
765 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBufferToImage() local
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1914 std::shared_ptr<mkldnn::memory> src_mem; member
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc1638 DeviceMemoryBase src_mem = DeviceMemoryBase(src_ptr, cur_stride_size); in ReorganizeMemory() local
1655 DeviceMemoryBase src_mem = DeviceMemoryBase(src_ptr, cur_stride_size); in ReorganizeMemory() local