Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_softmax_op.cc64 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data), in Execute()
69 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data)); in Execute()
78 context_.dst_mem->set_data_handle(DummyData); in Execute()
89 std::shared_ptr<memory> dst_mem; member
106 dst_mem(nullptr), in SoftmaxFwdContext()
129 context_.dst_mem.reset( in Setup()
135 {MKLDNN_ARG_DST, *context_.dst_mem}}); in Setup()
Dmkl_quantize_op.cc92 context_.dst_mem->set_data_handle(dst_data, *reorder_stream); in Execute()
95 context_.dst_mem->set_data_handle(dst_data); in Execute()
100 context_.dst_mem->set_data_handle(DummyData); in Execute()
108 std::shared_ptr<mkldnn::memory> dst_mem; member
121 dst_mem(nullptr), in ReorderContext()
131 context_.dst_mem.reset( in Setup()
146 context_.dst_mem->get_desc(), post_ops_attr)); in Setup()
151 context_.prim_args.insert({MKLDNN_ARG_TO, *context_.dst_mem}); in Setup()
Dmkl_slice_op.cc191 context_.dst_mem->set_data_handle(sliceParams.to->get_data_handle(), in Execute()
195 context_.dst_mem->set_data_handle(sliceParams.to->get_data_handle()); in Execute()
206 context_.dst_mem->set_data_handle(DummyData); in Execute()
215 std::shared_ptr<mkldnn::memory> dst_mem; member
223 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {} in SliceContext()
231 context_.dst_mem.reset( in Setup()
238 reorder::primitive_desc(*context_.src_sub_mem, *context_.dst_mem)); in Setup()
244 {MKLDNN_ARG_DST, *context_.dst_mem}}); in Setup()
Dmkl_pooling_ops_common.cc64 context_.dst_mem.reset( in Setup()
73 {MKLDNN_ARG_DST, *context_.dst_mem}, in Setup()
78 {MKLDNN_ARG_DST, *context_.dst_mem}}); in Setup()
92 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data), *fwd_stream); in Execute()
102 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data)); in Execute()
114 context_.dst_mem->set_data_handle(DummyData); in Execute()
Dmkl_fused_batch_norm_op.cc85 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data), in Execute()
105 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data)); in Execute()
125 context_.dst_mem->set_data_handle(DummyData); in Execute()
141 memory::desc GetDstPd() const { return context_.dst_mem->get_desc(); } in GetDstPd()
159 std::shared_ptr<mkldnn::memory> dst_mem; member
178 dst_mem(nullptr), in BatchNormFwdContext()
208 context_.dst_mem.reset( in Setup()
241 {MKLDNN_ARG_DST, *context_.dst_mem}}); in Setup()
244 {MKLDNN_ARG_DST, *context_.dst_mem}}); in Setup()
255 {MKLDNN_ARG_DST, *context_.dst_mem}, in Setup()
[all …]
Dmkl_matmul_ops_common.h111 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data), in Execute()
120 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data)); in Execute()
129 context_.dst_mem->set_data_handle(DummyData); in Execute()
144 std::shared_ptr<mkldnn::memory> dst_mem; member
166 dst_mem(nullptr), in MklDnnMatMulFwdContext()
269 context_.dst_mem.reset( in Setup()
281 {MKLDNN_ARG_DST, *context_.dst_mem}}); in Setup()
Dmkl_concat_op.cc287 context_.dst_mem->set_data_handle( in Execute()
293 context_.dst_mem->set_data_handle(
305 context_.dst_mem->set_data_handle(DummyData);
321 std::shared_ptr<mkldnn::memory> dst_mem; member
336 : dst_mem(nullptr), fwd_pd(nullptr), concat_fwd(nullptr) {} in ConcatFwdContext()
362 context_.dst_mem.reset( in Setup()
367 {MKLDNN_ARG_DST, *context_.dst_mem}}; in Setup()
Dmkl_relu_op.cc80 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data), in Execute()
85 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data)); in Execute()
94 context_.dst_mem->set_data_handle(DummyData); in Execute()
104 std::shared_ptr<memory> dst_mem; member
126 dst_mem(nullptr), in EltwiseFwdContext()
151 context_.dst_mem.reset( in Setup()
157 {MKLDNN_ARG_DST, *context_.dst_mem}}); in Setup()
Dmkl_conv_ops.cc127 context_.dst_mem->set_data_handle( in Execute()
138 context_.dst_mem->set_data_handle( in Execute()
155 context_.dst_mem->set_data_handle(DummyData); in Execute()
178 std::shared_ptr<mkldnn::memory> dst_mem; member
200 dst_mem(nullptr), in ConvFwdContext()
290 context_.dst_mem.reset( in Setup()
303 {MKLDNN_ARG_DST, *context_.dst_mem}}); in Setup()
309 {MKLDNN_ARG_DST, *context_.dst_mem}}); in Setup()
Dmkl_pooling_ops_common.h119 std::shared_ptr<mkldnn::memory> dst_mem; member
142 dst_mem(nullptr), in PoolingFwdContext()
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc111 void *dst_mem = gpu_dst->opaque(); in Memcpy() local
115 [dst_mem, host_src, size]() { memcpy(dst_mem, host_src, size); }); in Memcpy()
123 void *dst_mem = gpu_dst->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.cpp157 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/
Dmkl_util.h1231 const memory& dst_mem, const engine& engine,
1236 net_args.push_back({{MKLDNN_ARG_FROM, src_mem}, {MKLDNN_ARG_TO, dst_mem}});
1875 context_.dst_mem->set_data_handle(to->get_data_handle()); in SetMemory()
1883 std::shared_ptr<mkldnn::memory> dst_mem; member
1886 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {} in ReorderContext()
1894 context_.dst_mem.reset(new memory(to->get_desc(), cpu_engine_, DummyData)); in Setup()
1896 reorder(*context_.src_mem, *context_.dst_mem)); in Setup()
/external/mesa3d/src/gallium/auxiliary/hud/
Dfont.c357 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/zstd/contrib/freestanding_lib/
Dfreestanding.py498 dst_mem = os.path.join(self._dst_lib, "common", "mem.h")
500 shutil.copyfile(self._mem, dst_mem)