Home
last modified time | relevance | path

Searched refs:src_mem (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_softmax_op.cc62 context_.src_mem->set_data_handle( in Execute()
67 context_.src_mem->set_data_handle( in Execute()
77 context_.src_mem->set_data_handle(DummyData); in Execute()
88 std::shared_ptr<memory> src_mem; member
105 : src_mem(nullptr), in SoftmaxFwdContext()
127 context_.src_mem.reset( in Setup()
134 context_.fwd_net_args.push_back({{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
Dmkl_quantize_op.cc91 context_.src_mem->set_data_handle(src_data, *reorder_stream); in Execute()
94 context_.src_mem->set_data_handle(src_data); in Execute()
99 context_.src_mem->set_data_handle(DummyData); in Execute()
107 std::shared_ptr<mkldnn::memory> src_mem; member
120 : src_mem(nullptr), in ReorderContext()
129 context_.src_mem.reset( in Setup()
145 new ReorderPd(cpu_engine_, context_.src_mem->get_desc(), cpu_engine_, in Setup()
150 context_.prim_args.insert({MKLDNN_ARG_FROM, *context_.src_mem}); in Setup()
Dmkl_slice_op.cc189 context_.src_mem->set_data_handle(sliceParams.from->get_data_handle(), in Execute()
194 context_.src_mem->set_data_handle(sliceParams.from->get_data_handle()); in Execute()
205 context_.src_mem->set_data_handle(DummyData); in Execute()
214 std::shared_ptr<mkldnn::memory> src_mem; member
223 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {} in SliceContext()
229 context_.src_mem.reset( in Setup()
234 auto src_sub_desc = context_.src_mem->get_desc().submemory_desc( in Setup()
243 {{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
Dmkl_fused_batch_norm_op.cc83 context_.src_mem->set_data_handle( in Execute()
103 context_.src_mem->set_data_handle( in Execute()
124 context_.src_mem->set_data_handle(DummyData); in Execute()
157 std::shared_ptr<mkldnn::memory> src_mem; member
176 src_mem(nullptr), in BatchNormFwdContext()
206 context_.src_mem.reset( in Setup()
239 {{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
243 context_.net_args.push_back({{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
251 {{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
259 {{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
[all …]
Dmkl_pooling_ops_common.cc62 context_.src_mem.reset( in Setup()
72 context_.net_args.push_back({{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
77 context_.net_args.push_back({{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
90 context_.src_mem->set_data_handle( in Execute()
100 context_.src_mem->set_data_handle( in Execute()
113 context_.src_mem->set_data_handle(DummyData); in Execute()
Dmkl_relu_op.cc78 context_.src_mem->set_data_handle( in Execute()
83 context_.src_mem->set_data_handle( in Execute()
93 context_.src_mem->set_data_handle(DummyData); in Execute()
103 std::shared_ptr<memory> src_mem; member
125 : src_mem(nullptr), in EltwiseFwdContext()
149 context_.src_mem.reset( in Setup()
156 {{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
259 context_.src_mem->set_data_handle( in Execute()
266 context_.src_mem->set_data_handle( in Execute()
278 context_.src_mem->set_data_handle(DummyData); in Execute()
[all …]
Dmkl_conv_grad_filter_ops.cc109 context_.src_mem->set_data_handle( in Execute()
122 context_.src_mem->set_data_handle( in Execute()
136 context_.src_mem->set_data_handle(DummyData); in Execute()
163 std::shared_ptr<mkldnn::memory> src_mem; member
191 : src_mem(nullptr), in ConvBwdFilterContext()
259 context_.src_mem.reset( in Setup()
276 {{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
284 {{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
Dmkl_matmul_ops_common.h105 context_.src_mem->set_data_handle( in Execute()
114 context_.src_mem->set_data_handle( in Execute()
126 context_.src_mem->set_data_handle(DummyData); in Execute()
141 std::shared_ptr<mkldnn::memory> src_mem; member
163 : src_mem(nullptr), in MklDnnMatMulFwdContext()
265 context_.src_mem.reset( in Setup()
278 context_.net_args.push_back({{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
Dmkl_conv_ops.cc119 context_.src_mem->set_data_handle( in Execute()
130 context_.src_mem->set_data_handle( in Execute()
150 context_.src_mem->set_data_handle(DummyData); in Execute()
175 std::shared_ptr<mkldnn::memory> src_mem; member
197 : src_mem(nullptr), in ConvFwdContext()
286 context_.src_mem.reset( in Setup()
300 {{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
307 {{MKLDNN_ARG_SRC, *context_.src_mem}, in Setup()
Dmkl_pooling_ops_common.h118 std::shared_ptr<mkldnn::memory> src_mem; member
141 src_mem(nullptr), in PoolingFwdContext()
Dmkl_concat_op.cc347 std::shared_ptr<mkldnn::memory> src_mem( in Setup() local
349 context_.data_mem_shdptr.push_back(src_mem); in Setup()
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc103 void *src_mem = const_cast<void *>(gpu_src.opaque()); in Memcpy() local
105 [host_dst, src_mem, size]() { memcpy(host_dst, src_mem, size); }); in Memcpy()
124 void *src_mem = const_cast<void *>(gpu_src.opaque()); in MemcpyDeviceToDevice() local
129 [src_mem, dst_mem, size]() { memcpy(dst_mem, src_mem, size); }); in MemcpyDeviceToDevice()
/external/mesa3d/src/gallium/frontends/clover/api/
Dtransfer.cpp159 buffer &src_mem, in validate_copy() argument
162 if (dst_mem == src_mem) { in validate_copy()
518 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBuffer() local
529 validate_object(q, src_mem, src_origin, src_pitch, region); in clEnqueueCopyBuffer()
531 src_mem, src_origin, src_pitch, region); in clEnqueueCopyBuffer()
536 &src_mem, src_origin, region)); in clEnqueueCopyBuffer()
555 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBufferRect() local
566 validate_object(q, src_mem, src_origin, src_pitch, region); in clEnqueueCopyBufferRect()
568 src_mem, src_origin, src_pitch, region); in clEnqueueCopyBufferRect()
573 &src_mem, src_origin, src_pitch, in clEnqueueCopyBufferRect()
[all …]
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1230 const memory& src_mem,
1236 net_args.push_back({{MKLDNN_ARG_FROM, src_mem}, {MKLDNN_ARG_TO, dst_mem}});
1874 context_.src_mem->set_data_handle(from->get_data_handle()); in SetMemory()
1882 std::shared_ptr<mkldnn::memory> src_mem; member
1886 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {} in ReorderContext()
1892 context_.src_mem.reset( in Setup()
1896 reorder(*context_.src_mem, *context_.dst_mem)); in Setup()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc1953 DeviceMemoryBase src_mem = DeviceMemoryBase(src_ptr, cur_stride_size); in ReorganizeMemory() local
1957 ? stream->ThenMemcpy(&target_mem, src_mem, cur_stride_size).ok() in ReorganizeMemory()
1958 : stream->ThenMemcpy(&src_mem, target_mem, cur_stride_size).ok(); in ReorganizeMemory()
1970 DeviceMemoryBase src_mem = DeviceMemoryBase(src_ptr, cur_stride_size); in ReorganizeMemory() local
1973 gather ? stream->ThenMemcpy(&target_mem, src_mem, cur_stride_size).ok() in ReorganizeMemory()
1974 : stream->ThenMemcpy(&src_mem, target_mem, cur_stride_size).ok(); in ReorganizeMemory()