| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 78 void *dst_mem = gpu_dst->opaque(); in Memcpy() local
82 [dst_mem, host_src, size]() { memcpy(dst_mem, host_src, size); }); in Memcpy()
90 void *dst_mem = gpu_dst->opaque(); in MemcpyDeviceToDevice() local
96 [src_mem, dst_mem, size]() { memcpy(dst_mem, src_mem, size); }); in MemcpyDeviceToDevice()
|
| /external/mesa3d/src/gallium/state_trackers/clover/api/ |
| D | transfer.cpp | 135 validate_copy(command_queue &q, buffer &dst_mem, in validate_copy() argument
140 if (dst_mem == src_mem) { in validate_copy()
428 auto &dst_mem = obj<buffer>(d_dst_mem); in clEnqueueCopyBuffer() local
437 validate_object(q, dst_mem, dst_origin, dst_pitch, region); in clEnqueueCopyBuffer()
439 validate_copy(q, dst_mem, dst_origin, dst_pitch, in clEnqueueCopyBuffer()
444 hard_copy_op(q, &dst_mem, dst_origin, in clEnqueueCopyBuffer()
465 auto &dst_mem = obj<buffer>(d_dst_mem); in clEnqueueCopyBufferRect() local
474 validate_object(q, dst_mem, dst_origin, dst_pitch, region); in clEnqueueCopyBufferRect()
476 validate_copy(q, dst_mem, dst_origin, dst_pitch, in clEnqueueCopyBufferRect()
481 soft_copy_op(q, &dst_mem, dst_origin, dst_pitch, in clEnqueueCopyBufferRect()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_slice_op.cc | 190 context_.dst_mem->set_data_handle(sliceParams.to->get_data_handle()); in Execute()
198 context_.dst_mem->set_data_handle(DummyData); in Execute()
207 std::shared_ptr<mkldnn::memory> dst_mem; member
214 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {} in SliceContext()
225 context_.dst_mem.reset(new memory( in Setup()
228 auto dst_pd = context_.dst_mem->get_primitive_desc(); in Setup()
236 reorder(*context_.reorder_pd, *context_.src_mem, *context_.dst_mem)); in Setup()
|
| D | mkl_fused_batch_norm_op.cc | 67 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data)); in Execute()
83 context_.dst_mem->set_data_handle(DummyData); in Execute()
96 return (*context_.dst_mem).get_primitive_desc(); in GetDstPd()
104 return (*context_.dst_mem).get_primitive_desc().desc().data.format; in GetDstFmt()
119 std::shared_ptr<mkldnn::memory> dst_mem; member
133 dst_mem(nullptr), in BatchNormFwdContext()
158 context_.dst_mem.reset(new memory(fwd_pd.dst_primitive_desc(), DummyData)); in Setup()
183 *context_.dst_mem)); in Setup()
186 fwd_pd, *context_.src_mem, *context_.dst_mem)); in Setup()
193 *context_.dst_mem)); in Setup()
[all …]
|
| D | mkl_pooling_ops_common.cc | 78 context_.dst_mem.reset( in Setup()
94 *context_.dst_mem, in Setup()
98 *context_.dst_mem)); in Setup()
109 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data)); in Execute()
120 context_.dst_mem->set_data_handle(DummyData); in Execute()
|
| D | mkl_relu_op.cc | 82 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data)); in Execute()
87 context_.dst_mem->set_data_handle(DummyData); in Execute()
104 std::shared_ptr<memory> dst_mem; member
126 dst_mem(nullptr), in EltwiseFwdContext()
152 context_.dst_mem.reset( in Setup()
157 *context_.fwd_pd, *context_.src_mem, *context_.dst_mem)); in Setup()
|
| D | mkl_conv_ops.cc | 128 context_.dst_mem->set_data_handle( in Execute()
136 context_.dst_mem->set_data_handle(DummyData); in Execute()
151 context_.dst_mem->set_data_handle( in Execute()
158 context_.dst_mem->set_data_handle(DummyData); in Execute()
180 std::shared_ptr<mkldnn::memory> dst_mem; member
204 dst_mem(nullptr), in ConvFwdContext()
293 context_.dst_mem.reset( in Setup()
304 *context_.bias_mem, *context_.dst_mem)); in Setup()
308 *context_.filter_mem, *context_.dst_mem)); in Setup()
|
| D | mkl_pooling_ops_common.h | 118 std::shared_ptr<mkldnn::memory> dst_mem; member
139 dst_mem(nullptr), in PoolingFwdContext()
|
| /external/mesa3d/src/gallium/auxiliary/hud/ |
| D | font.c | 357 util_font_draw_character(void *dst_mem, unsigned dst_stride, unsigned character) in util_font_draw_character() argument
359 unsigned char *dst = (unsigned char*)dst_mem; in util_font_draw_character()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 2269 context_.dst_mem->set_data_handle(to->get_data_handle()); in SetMemory()
2275 std::shared_ptr<mkldnn::memory> dst_mem; member
2278 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {} in ReorderContext()
2286 context_.dst_mem.reset( in Setup()
2289 reorder(*context_.src_mem, *context_.dst_mem)); in Setup()
|