| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_softmax_op.cc | 62 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data)); in Execute()
68 context_.dst_mem->set_data_handle(DummyData); in Execute()
79 std::shared_ptr<memory> dst_mem; member
96 dst_mem(nullptr), in SoftmaxFwdContext()
119 context_.dst_mem.reset( in Setup()
124 *context_.fwd_pd, *context_.src_mem, *context_.dst_mem)); in Setup()
|
| D | mkl_slice_op.cc | 189 context_.dst_mem->set_data_handle(sliceParams.to->get_data_handle()); in Execute()
197 context_.dst_mem->set_data_handle(DummyData); in Execute()
206 std::shared_ptr<mkldnn::memory> dst_mem; member
213 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {} in SliceContext()
224 context_.dst_mem.reset(new memory( in Setup()
227 auto dst_pd = context_.dst_mem->get_primitive_desc(); in Setup()
235 reorder(*context_.reorder_pd, *context_.src_mem, *context_.dst_mem)); in Setup()
|
| D | mkl_matmul_ops_common.h | 96 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data)); in Execute()
103 context_.dst_mem->set_data_handle(DummyData); in Execute()
124 std::shared_ptr<mkldnn::memory> dst_mem; member
147 dst_mem(nullptr), in MklDnnMatMulFwdContext()
240 context_.dst_mem.reset( in Setup()
251 *context_.bias_mem, *context_.dst_mem)); in Setup()
|
| D | mkl_fused_batch_norm_op.cc | 72 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data)); in Execute()
88 context_.dst_mem->set_data_handle(DummyData); in Execute()
101 return (*context_.dst_mem).get_primitive_desc(); in GetDstPd()
109 return (*context_.dst_mem).get_primitive_desc().desc().data.format; in GetDstFmt()
124 std::shared_ptr<mkldnn::memory> dst_mem; member
138 dst_mem(nullptr), in BatchNormFwdContext()
163 context_.dst_mem.reset(new memory(fwd_pd.dst_primitive_desc(), DummyData)); in Setup()
188 *context_.dst_mem)); in Setup()
191 fwd_pd, *context_.src_mem, *context_.dst_mem)); in Setup()
198 *context_.dst_mem)); in Setup()
[all …]
|
| D | mkl_pooling_ops_common.cc | 77 context_.dst_mem.reset( in Setup()
93 *context_.dst_mem, in Setup()
97 *context_.dst_mem)); in Setup()
108 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data)); in Execute()
119 context_.dst_mem->set_data_handle(DummyData); in Execute()
|
| D | mkl_quantize_op.cc | 94 context_.dst_mem->set_data_handle(to->get_data_handle()); in SetMemory()
102 std::shared_ptr<mkldnn::memory> dst_mem; member
118 dst_mem(nullptr), in ReorderContext()
157 context_.dst_mem.reset(new memory(*context_.dst_mpd, DummyData)); in Setup()
161 reorder(*context_.reorder_pd, *context_.src_mem, *context_.dst_mem)); in Setup()
|
| D | mkl_concat_op.cc | 286 context_.dst_mem->set_data_handle( in Execute()
296 context_.dst_mem->set_data_handle(DummyData); in Execute()
316 std::shared_ptr<mkldnn::memory> dst_mem; member
330 : dst_mem(nullptr), in ConcatFwdContext()
369 context_.dst_mem.reset(new memory(*context_.dst_pd, DummyData)); in Setup()
373 new concat(*context_.fwd_pd, context_.data_mem, *context_.dst_mem)); in Setup()
|
| 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 | 124 context_.dst_mem->set_data_handle( in Execute()
143 context_.dst_mem->set_data_handle(DummyData); in Execute()
179 std::shared_ptr<mkldnn::memory> dst_mem; member
210 dst_mem(nullptr), in ConvFwdContext()
308 context_.dst_mem.reset(new MEMORY_CONSTRUCTOR( in Setup()
323 *context_.dst_mem }}); in Setup()
330 *context_.dst_mem }}); in Setup()
335 *context_.bias_mem, *context_.dst_mem)); in Setup()
339 *context_.filter_mem, *context_.dst_mem));
|
| D | mkl_pooling_ops_common.h | 117 std::shared_ptr<mkldnn::memory> dst_mem; member
138 dst_mem(nullptr), in PoolingFwdContext()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 87 void *dst_mem = gpu_dst->opaque(); in Memcpy() local
91 [dst_mem, host_src, size]() { memcpy(dst_mem, host_src, size); }); in Memcpy()
99 void *dst_mem = gpu_dst->opaque(); in MemcpyDeviceToDevice() local
105 [src_mem, dst_mem, size]() { memcpy(dst_mem, src_mem, size); }); in MemcpyDeviceToDevice()
|
| /external/mesa3d/src/gallium/frontends/clover/api/ |
| D | transfer.cpp | 157 validate_copy(command_queue &q, buffer &dst_mem, in validate_copy() argument
162 if (dst_mem == src_mem) { in validate_copy()
519 auto &dst_mem = obj<buffer>(d_dst_mem); in clEnqueueCopyBuffer() local
528 validate_object(q, dst_mem, dst_origin, dst_pitch, region); in clEnqueueCopyBuffer()
530 validate_copy(q, dst_mem, dst_origin, dst_pitch, in clEnqueueCopyBuffer()
535 hard_copy_op(q, &dst_mem, dst_origin, in clEnqueueCopyBuffer()
556 auto &dst_mem = obj<buffer>(d_dst_mem); in clEnqueueCopyBufferRect() local
565 validate_object(q, dst_mem, dst_origin, dst_pitch, region); in clEnqueueCopyBufferRect()
567 validate_copy(q, dst_mem, dst_origin, dst_pitch, in clEnqueueCopyBufferRect()
572 soft_copy_op(q, &dst_mem, dst_origin, dst_pitch, in clEnqueueCopyBufferRect()
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1281 const memory& dst_mem, in CreateAndExecuteReorder() argument
1287 net_args.push_back({{MKLDNN_ARG_FROM, src_mem}, {MKLDNN_ARG_TO, dst_mem}}); in CreateAndExecuteReorder()
1289 net.push_back(mkldnn::reorder(reorder_desc, src_mem, dst_mem)); in CreateAndExecuteReorder()
2033 context_.dst_mem->set_data_handle(to->get_data_handle());
2039 std::shared_ptr<mkldnn::memory> dst_mem; member
2042 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {}
2050 context_.dst_mem.reset(
2053 reorder(*context_.src_mem, *context_.dst_mem));
|
| /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()
|