| /external/tensorflow/tensorflow/stream_executor/host/ |
| host_gpu_executor.cc | 70 void *src_mem = const_cast<void *>(gpu_src.opaque()); local
72 [host_dst, src_mem, size]() { memcpy(host_dst, src_mem, size); });
91 void *src_mem = const_cast<void *>(gpu_src.opaque()); local
96 [src_mem, dst_mem, size]() { memcpy(dst_mem, src_mem, size); });
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_slice_op.cc | 189 context_.src_mem->set_data_handle(sliceParams.from->get_data_handle());
197 context_.src_mem->set_data_handle(DummyData);
206 std::shared_ptr<mkldnn::memory> src_mem; member in struct:tensorflow::MklSlicePrimitive::SliceContext
214 : src_mem(nullptr), dst_mem(nullptr), reorder_prim(nullptr) {}
222 context_.src_mem.reset(
227 auto src_pd = context_.src_mem->get_primitive_desc();
236 reorder(*context_.reorder_pd, *context_.src_mem, *context_.dst_mem));
|
| mkl_conv_grad_filter_ops.cc | 100 context_.src_mem->set_data_handle(
111 context_.src_mem->set_data_handle(DummyData);
124 context_.src_mem->set_data_handle(
133 context_.src_mem->set_data_handle(DummyData);
169 std::shared_ptr<mkldnn::memory> src_mem; member in struct:tensorflow::MklConvBwdFilterPrimitive::ConvBwdFilterContext
193 src_mem(nullptr),
267 context_.src_mem.reset(
283 *context_.bwd_filter_pd, *context_.src_mem, *context_.diff_dst_mem,
287 *context_.bwd_filter_pd, *context_.src_mem, *context_.diff_dst_mem,
|
| mkl_pooling_ops_common.h | 117 std::shared_ptr<mkldnn::memory> src_mem; member in struct:tensorflow::MklPoolingFwdPrimitive::PoolingFwdContext
138 src_mem(nullptr),
|
| mkl_relu_op.cc | 80 context_.src_mem->set_data_handle(
86 context_.src_mem->set_data_handle(DummyData);
103 std::shared_ptr<memory> src_mem; member in struct:tensorflow::MklEltwiseFwdPrimitive::EltwiseFwdContext
125 src_mem(nullptr),
151 context_.src_mem.reset(new memory(*context_.src_mpd, DummyData));
157 *context_.fwd_pd, *context_.src_mem, *context_.dst_mem));
265 context_.src_mem->set_data_handle(
273 context_.src_mem->set_data_handle(DummyData);
294 std::shared_ptr<memory> src_mem; member in struct:tensorflow::MklEltwiseBwdPrimitive::EltwiseBwdContext
324 src_mem(nullptr)
[all...] |
| mkl_fused_batch_norm_op.cc | 65 context_.src_mem->set_data_handle(
82 context_.src_mem->set_data_handle(DummyData);
100 return (*context_.src_mem).get_primitive_desc().desc().data.format;
117 std::shared_ptr<mkldnn::memory> src_mem; member in struct:tensorflow::MklFusedBatchNormFwdPrimitive::BatchNormFwdContext
131 src_mem(nullptr),
157 context_.src_mem.reset(new memory({src_md, cpu_engine_}, DummyData));
182 fwd_pd, *context_.src_mem, *context_.weights_mem,
186 fwd_pd, *context_.src_mem, *context_.dst_mem));
191 fwd_pd, *context_.src_mem, (const primitive::at)*context_.mean_mem,
196 fwd_pd, *context_.src_mem, (const primitive::at)*context_.mean_mem
359 std::shared_ptr<mkldnn::memory> src_mem; member in struct:tensorflow::MklFusedBatchNormBwdPrimitive::BatchNormBwdContext
[all...] |
| mkl_conv_ops.cc | 122 context_.src_mem->set_data_handle(
133 context_.src_mem->set_data_handle(DummyData);
147 context_.src_mem->set_data_handle(
156 context_.src_mem->set_data_handle(DummyData);
177 std::shared_ptr<mkldnn::memory> src_mem; member in struct:tensorflow::MklConvFwdPrimitive::ConvFwdContext
201 src_mem(nullptr),
289 context_.src_mem.reset(
303 *context_.fwd_pd, *context_.src_mem, *context_.filter_mem,
307 new convolution_forward(*context_.fwd_pd, *context_.src_mem,
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| mkl_util.h | 2274 std::shared_ptr<mkldnn::memory> src_mem; member in struct:MklReorderPrimitive::ReorderContext
[all...] |
