Home
last modified time | relevance | path

Searched refs:GetOpMem (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_lrn_op.cc184 net.push_back(lrn_forward(lrn_fwd_desc, src_dnn_data->GetOpMem(), in PrepareAndExecuteNet()
185 wksp_dnn_data->GetOpMem(), in PrepareAndExecuteNet()
186 dst_dnn_data->GetOpMem())); in PrepareAndExecuteNet()
188 net.push_back(lrn_forward(lrn_fwd_desc, src_dnn_data->GetOpMem(), in PrepareAndExecuteNet()
189 dst_dnn_data->GetOpMem())); in PrepareAndExecuteNet()
499 net.push_back(lrn_backward(lrn_bkwd_desc, src_dnn_data->GetOpMem(), in PrepareAndExecuteNet()
500 input_gradient_diff_dst->GetOpMem(), in PrepareAndExecuteNet()
501 output_diff_src->GetOpMem())); in PrepareAndExecuteNet()
503 net.push_back(lrn_backward(lrn_bkwd_desc, src_dnn_data->GetOpMem(), in PrepareAndExecuteNet()
504 input_gradient_diff_dst->GetOpMem(), in PrepareAndExecuteNet()
[all …]
Dmkl_aggregate_ops.cc209 inputs.push_back(src1.GetOpMem()); in Compute()
210 inputs.push_back(src2.GetOpMem()); in Compute()
242 net.push_back(sum(sum_pd, inputs, dst.GetOpMem())); in Compute()
Dmkl_softmax_op.cc165 softmax_forward(softmax_fwd_pd, src.GetOpMem(), dst.GetOpMem()); in Compute()
Dmkl_conv_ops.cc1104 src_data = static_cast<Tinput*>(src.GetOpMem().get_data_handle()); in Compute()
1138 static_cast<Tfilter*>(filter.GetOpMem().get_data_handle()); in Compute()
1378 net.push_back(convolution_forward(conv_prim_desc, src->GetOpMem(), in PrepareAndExecuteNet()
1379 filter->GetOpMem(), bias->GetOpMem(), in PrepareAndExecuteNet()
1380 output->GetOpMem())); in PrepareAndExecuteNet()
1383 net.push_back(convolution_forward(conv_prim_desc, src->GetOpMem(), in PrepareAndExecuteNet()
1384 filter->GetOpMem(), in PrepareAndExecuteNet()
1385 output->GetOpMem())); in PrepareAndExecuteNet()
1420 filter_data = static_cast<Tfilter*>(filter.GetOpMem().get_data_handle()); in CacheFilter()
1426 filter.GetOpMem().get_primitive_desc().get_size(); in CacheFilter()
Dmkl_maxpooling_op.cc604 reinterpret_cast<T*>(dnn_data_input.GetOpMem().get_data_handle())); in Compute()
636 static_cast<T*>(dnn_data_wksp.GetOpMem().get_data_handle()); in Compute()
760 reinterpret_cast<T*>(grad_dnn_data.GetOpMem().get_data_handle())); in Compute()
778 ws_data = workspace_dnn_data.GetOpMem().get_data_handle(); in Compute()
Dmkl_conv_grad_filter_ops.cc542 src_data = static_cast<T*>(src.GetOpMem().get_data_handle()); in Compute()
552 diff_dst_data = static_cast<T*>(diff_dst.GetOpMem().get_data_handle()); in Compute()
571 static_cast<T*>(diff_filter.GetOpMem().get_data_handle()); in Compute()
Dmkl_conv_grad_input_ops.cc444 filter_data = static_cast<T*>(filter.GetOpMem().get_data_handle()); in Compute()
454 diff_dst_data = static_cast<T*>(diff_dst.GetOpMem().get_data_handle()); in Compute()
Dmkl_concat_op.cc377 inputs.push_back(srcs[k].GetOpMem()); in Compute()
417 auto concat_op = concat(concat_pd, inputs, dst.GetOpMem()); in Compute()
Dmkl_avgpooling_op.cc527 reinterpret_cast<T*>(dnn_data_input.GetOpMem().get_data_handle())); in Compute()
649 reinterpret_cast<T*>(grad_dnn_data.GetOpMem().get_data_handle())); in Compute()
Dmkl_fused_batch_norm_op.cc655 reinterpret_cast<T*>(src.GetOpMem().get_data_handle())); in Compute()
972 reinterpret_cast<T*>(src.GetOpMem().get_data_handle())); in Compute()
985 reinterpret_cast<T*>(diff_dst.GetOpMem().get_data_handle())); in Compute()
Dmkl_relu_op.cc499 reinterpret_cast<T*>(src.GetOpMem().get_data_handle())); in Compute()
654 reinterpret_cast<T*>(src.GetOpMem().get_data_handle())); in Compute()
663 reinterpret_cast<T*>(diff_dst.GetOpMem().get_data_handle())); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1836 inline const memory& GetOpMem() const { in GetOpMem() function