| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 70 void *src_mem = const_cast<void *>(gpu_src.opaque()); in Memcpy() local
72 [host_dst, src_mem, size]() { memcpy(host_dst, src_mem, size); }); in Memcpy()
91 void *src_mem = const_cast<void *>(gpu_src.opaque()); in MemcpyDeviceToDevice() local
96 [src_mem, dst_mem, size]() { memcpy(dst_mem, src_mem, size); }); in MemcpyDeviceToDevice()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_fused_batch_norm_op.cc | 65 context_.src_mem->set_data_handle( in Execute()
82 context_.src_mem->set_data_handle(DummyData); in Execute()
100 return (*context_.src_mem).get_primitive_desc().desc().data.format; in GetSrcFmt()
117 std::shared_ptr<mkldnn::memory> src_mem; member
131 src_mem(nullptr), in BatchNormFwdContext()
157 context_.src_mem.reset(new memory({src_md, cpu_engine_}, DummyData)); in Setup()
182 fwd_pd, *context_.src_mem, *context_.weights_mem, in Setup()
186 fwd_pd, *context_.src_mem, *context_.dst_mem)); in Setup()
191 fwd_pd, *context_.src_mem, (const primitive::at)*context_.mean_mem, in Setup()
196 fwd_pd, *context_.src_mem, (const primitive::at)*context_.mean_mem, in Setup()
[all …]
|
| D | mkl_slice_op.cc | 189 context_.src_mem->set_data_handle(sliceParams.from->get_data_handle()); in Execute()
197 context_.src_mem->set_data_handle(DummyData); in Execute()
206 std::shared_ptr<mkldnn::memory> src_mem; member
214 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {} in SliceContext()
222 context_.src_mem.reset( in Setup()
227 auto src_pd = context_.src_mem->get_primitive_desc(); in Setup()
236 reorder(*context_.reorder_pd, *context_.src_mem, *context_.dst_mem)); in Setup()
|
| D | mkl_conv_grad_filter_ops.cc | 100 context_.src_mem->set_data_handle( in Execute()
111 context_.src_mem->set_data_handle(DummyData); in Execute()
124 context_.src_mem->set_data_handle( in Execute()
133 context_.src_mem->set_data_handle(DummyData); in Execute()
169 std::shared_ptr<mkldnn::memory> src_mem; member
193 src_mem(nullptr), in ConvBwdFilterContext()
267 context_.src_mem.reset( in Setup()
283 *context_.bwd_filter_pd, *context_.src_mem, *context_.diff_dst_mem, in Setup()
287 *context_.bwd_filter_pd, *context_.src_mem, *context_.diff_dst_mem, in Setup()
|
| D | mkl_relu_op.cc | 80 context_.src_mem->set_data_handle( in Execute()
86 context_.src_mem->set_data_handle(DummyData); in Execute()
103 std::shared_ptr<memory> src_mem; member
125 src_mem(nullptr), in EltwiseFwdContext()
151 context_.src_mem.reset(new memory(*context_.src_mpd, DummyData)); in Setup()
157 *context_.fwd_pd, *context_.src_mem, *context_.dst_mem)); in Setup()
265 context_.src_mem->set_data_handle( in Execute()
273 context_.src_mem->set_data_handle(DummyData); in Execute()
294 std::shared_ptr<memory> src_mem; member
324 src_mem(nullptr), in EltwiseBwdContext()
[all …]
|
| D | mkl_pooling_ops_common.cc | 75 context_.src_mem.reset(new memory( in Setup()
93 context_.fwd.reset(new pooling_forward(*context_.fwd_pd, *context_.src_mem, in Setup()
97 context_.fwd.reset(new pooling_forward(*context_.fwd_pd, *context_.src_mem, in Setup()
107 context_.src_mem->set_data_handle( in Execute()
119 context_.src_mem->set_data_handle(DummyData); in Execute()
|
| D | mkl_conv_ops.cc | 122 context_.src_mem->set_data_handle( in Execute()
133 context_.src_mem->set_data_handle(DummyData); in Execute()
147 context_.src_mem->set_data_handle( in Execute()
156 context_.src_mem->set_data_handle(DummyData); in Execute()
177 std::shared_ptr<mkldnn::memory> src_mem; member
201 src_mem(nullptr), in ConvFwdContext()
289 context_.src_mem.reset( in Setup()
303 *context_.fwd_pd, *context_.src_mem, *context_.filter_mem, in Setup()
307 new convolution_forward(*context_.fwd_pd, *context_.src_mem, in Setup()
|
| D | mkl_pooling_ops_common.h | 117 std::shared_ptr<mkldnn::memory> src_mem; member
138 src_mem(nullptr), in PoolingFwdContext()
|
| /external/mesa3d/src/gallium/state_trackers/clover/api/ |
| D | transfer.cpp | 137 buffer &src_mem, in validate_copy() argument
140 if (dst_mem == src_mem) { in validate_copy()
427 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBuffer() local
438 validate_object(q, src_mem, src_origin, src_pitch, region); in clEnqueueCopyBuffer()
440 src_mem, src_origin, src_pitch, region); in clEnqueueCopyBuffer()
445 &src_mem, src_origin, region)); in clEnqueueCopyBuffer()
464 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBufferRect() local
475 validate_object(q, src_mem, src_origin, src_pitch, region); in clEnqueueCopyBufferRect()
477 src_mem, src_origin, src_pitch, region); in clEnqueueCopyBufferRect()
482 &src_mem, src_origin, src_pitch, in clEnqueueCopyBufferRect()
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 2268 context_.src_mem->set_data_handle(from->get_data_handle()); in SetMemory()
2274 std::shared_ptr<mkldnn::memory> src_mem; member
2278 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {} in ReorderContext()
2284 context_.src_mem.reset(new memory( in Setup()
2289 reorder(*context_.src_mem, *context_.dst_mem)); in Setup()
|