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_slice_op.cc214 std::shared_ptr<mkldnn::memory> src_mem; member
Dmkl_quantize_op.cc107 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.cc157 std::shared_ptr<mkldnn::memory> src_mem; member
492 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.h141 std::shared_ptr<mkldnn::memory> src_mem; member
Dmkl_concat_op.cc347 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.cc175 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.h1882 std::shared_ptr<mkldnn::memory> src_mem; member
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc1953 DeviceMemoryBase src_mem = DeviceMemoryBase(src_ptr, cur_stride_size); in ReorganizeMemory() local
1970 DeviceMemoryBase src_mem = DeviceMemoryBase(src_ptr, cur_stride_size); in ReorganizeMemory() local