| /external/tensorflow/tensorflow/stream_executor/host/ |
| host_gpu_executor.cc | 78 void *dst_mem = gpu_dst->opaque(); local
82 [dst_mem, host_src, size]() { memcpy(dst_mem, host_src, size); });
90 void *dst_mem = gpu_dst->opaque(); local
96 [src_mem, dst_mem, size]() { memcpy(dst_mem, src_mem, size); });
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_slice_op.cc | 190 context_.dst_mem->set_data_handle(sliceParams.to->get_data_handle());
198 context_.dst_mem->set_data_handle(DummyData);
207 std::shared_ptr<mkldnn::memory> dst_mem; member in struct:tensorflow::MklSlicePrimitive::SliceContext
214 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {}
225 context_.dst_mem.reset(new memory(
228 auto dst_pd = context_.dst_mem->get_primitive_desc();
236 reorder(*context_.reorder_pd, *context_.src_mem, *context_.dst_mem));
|
| mkl_pooling_ops_common.h | 118 std::shared_ptr<mkldnn::memory> dst_mem; member in struct:tensorflow::MklPoolingFwdPrimitive::PoolingFwdContext
139 dst_mem(nullptr),
|
| mkl_relu_op.cc | 82 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data));
87 context_.dst_mem->set_data_handle(DummyData);
104 std::shared_ptr<memory> dst_mem; member in struct:tensorflow::MklEltwiseFwdPrimitive::EltwiseFwdContext
126 dst_mem(nullptr),
152 context_.dst_mem.reset(
157 *context_.fwd_pd, *context_.src_mem, *context_.dst_mem));
[all...] |
| mkl_fused_batch_norm_op.cc | 67 context_.dst_mem->set_data_handle(static_cast<void*>(dst_data));
83 context_.dst_mem->set_data_handle(DummyData);
96 return (*context_.dst_mem).get_primitive_desc();
104 return (*context_.dst_mem).get_primitive_desc().desc().data.format;
119 std::shared_ptr<mkldnn::memory> dst_mem; member in struct:tensorflow::MklFusedBatchNormFwdPrimitive::BatchNormFwdContext
133 dst_mem(nullptr),
158 context_.dst_mem.reset(new memory(fwd_pd.dst_primitive_desc(), DummyData));
183 *context_.dst_mem));
186 fwd_pd, *context_.src_mem, *context_.dst_mem));
193 *context_.dst_mem));
[all...] |
| mkl_conv_ops.cc | 128 context_.dst_mem->set_data_handle(
136 context_.dst_mem->set_data_handle(DummyData);
151 context_.dst_mem->set_data_handle(
158 context_.dst_mem->set_data_handle(DummyData);
180 std::shared_ptr<mkldnn::memory> dst_mem; member in struct:tensorflow::MklConvFwdPrimitive::ConvFwdContext
204 dst_mem(nullptr),
293 context_.dst_mem.reset(
304 *context_.bias_mem, *context_.dst_mem));
308 *context_.filter_mem, *context_.dst_mem));
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| mkl_util.h | 2275 std::shared_ptr<mkldnn::memory> dst_mem; member in struct:MklReorderPrimitive::ReorderContext
[all...] |
